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' >> ~/.bashrcecho '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,原因未知。(可選)
- NumPy (~30s):
python -c "import numpy; numpy.test()"
- SciPy (~1m):
python -c "import scipy; scipy.test()"
- Theano (~30m):
python -c "import theano; theano.test()"
BACKUP YOUR SB UBUNTU IMMEDIATELY!
鳴謝: PKU-Esquirrel WIT506-Leo 深度學習大講堂微信公眾號及研討班的同學 參考: