1. 程式人生 > >Ubuntu 14.04.3 深度學習環境配置

Ubuntu 14.04.3 深度學習環境配置

機器型號:Lenovo M5400  顯示卡:NVIDIA GT740M  CPU:Intel 酷睿i5 4200M 記憶體:8G

(一)安裝Ubuntu 14.04.3

參考

(二)解除安裝系統自帶的顯示卡驅動nouveau

必須徹底解除安裝nouveau之後才能安裝NVIDIA驅動!否則驅動衝突會導致系統崩潰事故,只能重灌系統!

1. 開啟命令列,執行 sudo nautilus,以root許可權訪問系統目錄。

2. 進入資料夾 /lib/modules/3.19.0-25-generic/kernel/drivers/gpu/drm/nouveau,直接刪除裡邊的全部檔案。注意這些檔案只有root許可權才可以改動,所以需要第1步,否則可以使用chown指令修改檔案所有者之後再刪除。

3. 重新啟動計算機。

(三)安裝NVIDIA顯示卡驅動

2. 執行 sudo chmod +x NVIDIA-Linux-x86_64-375.26.run,賦予其可執行許可權。不同顯示卡的驅動名稱和版本不同,可以使用TAB自動補全。

3. 執行 sudo service lightdm stop,關閉Xwidnow。必須先關閉Xwindow,之後才能安裝NVIDIA驅動!否則將導致系統崩潰事故,只能重灌系統!

4. 按下Ctrl + Alt + F1 進入控制檯,登入後執行 sudo ./NVIDIA-Linux-x86_64-375.26.run,開始安裝NVIDIA驅動。其間會提示一個錯誤(無預檔案?),可以無視。

5. 安裝完成後,執行 sudo service lightdm start,啟動Xwindow。由於某些原因,此時輸入密碼登入使用者介面時可能會出現登入無限迴圈的情況!解決方法是安裝gnome的Flashback桌面!參考:Ubuntu 15.04 on vmware開機登入介面無限迴圈的解決辦法

6. 以上步驟完成後,重新啟動計算機,執行nvidia-smi,查詢驅動是否已經成功安裝。

[email protected]:~$ nvidia-smi
Sat Jan 28 20:02:32 2017       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.26                 Driver Version: 375.26                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GT 740M     Off  | 0000:01:00.0     N/A |                  N/A |
| N/A   43C    P0    N/A /  N/A |      0MiB /  2003MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|    0                  Not Supported                                         |
+-----------------------------------------------------------------------------+

提示如上資訊,即已成功安裝!

(四)安裝CUDA

步驟1~5與(三)完全對應相同。

2. 執行 sudo chmod +x cuda_8.0.44_linux.run,賦予其可執行許可權。可以使用TAB自動補全。

3. 執行 sudo service lightdm stop,關閉Xwidnow。不關閉Xwindow直接安裝CUDA的副作用尚不明確。

4. 按下Ctrl + Alt + F1 進入控制檯,登入後執行 sudo ./cuda_8.0.44_linux.run,開始安裝CUDA驅動。注意要恰當選擇安裝過程中的選項。由於已經安裝了NVIDIA顯示卡驅動,此時提示安裝驅動時已無需再次安裝。

5. 安裝完成後,執行 sudo service lightdm start,啟動Xwindow。

6. 設定環境變數。在命令列中輸入以下命令即可。

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
7. 以上步驟完成後,重新啟動計算機,執行nvcc -V,查詢驅動是否已經成功安裝。 [email protected]:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44
提示如上資訊,即已成功安裝! 至此,Ubuntu 14.04.3的深度學習環境必要依賴已配置成功,在此基礎上可以輕鬆安裝並執行Caffe,Tensorflow,Theano,Keras等深度學習框架。 (五)安裝Numpy,Scipy,Theano 1. 執行以下命令,注意第一個命令必須執行成功後再執行後面的,否則可能出錯。由於網速等原因可能無法一次更新成功,這時就要多更新幾次或者調整一下軟體源。 sudo apt-get update sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git sudo pip install Theano sudo pip install nose_parameterized 2. 安裝完成後執行以下命令測試,注意在執行Scipy的測試時有一定機率出現一個Error,原因未知。(可選)
  1. NumPy (~30s): python -c "import numpy; numpy.test()"
  2. SciPy (~1m): python -c "import scipy; scipy.test()"
  3. Theano (~30m): python -c "import theano; theano.test()"
(六)安裝Tensorflow 2. 執行命令安裝即可:sudo pip install tensorflow_gpu-0.12.1-cp27-none-linux_x86_64.whl (七)安裝Keras 1. 直接執行命令即可:sudo pip install keras NEVER! EVER! UPDATE THE GODDAMNSYSTEMOR ANY SOFTWARE!

BACKUP YOUR SB UBUNTU IMMEDIATELY!

鳴謝: PKU-Esquirrel WIT506-Leo 深度學習大講堂微信公眾號及研討班的同學 參考: