我們在安裝CUDA的時候要留意不同的ubuntu系統對應不同的版本;同時需要注意我們的ubuntu系統要選用64bit的,我在使用32bit的桌面ubuntu時,出現了CUDA無法正常使用的問題。
爲了使用CUDA,我們首先得安裝nVIDIA Driver驅動程式,然後安裝CUDA Toolkit工具;最後我們可以通過CUDA Samples來檢測CUDA是否可以正常工作。
具體的安裝步驟:
A.安裝nVIDIA Driver驅動
如果使用 cuda_xxxx.run 來安裝 nVIDIA driver,會出現Ubuntu 桌面無法開啓的情況,所以我們使用以下的步驟來安裝 nVIDIA driver。
1.首先安裝開發必須套件:
sudo apt-get install build-essential g++ vim
2.接着安裝 nVIDIA顯示卡 driver,安裝之前必須將 Ubuntu 的 x window關掉。切換到文字界面:
Ctrl+Alt+F1
3.關掉x window界面:
sudo service lightdm stop
4.添加驅動的下載源:
sudo add-apt-repository ppa:xorg-edgers/ppa
sudo apt-get update
5.安裝340版本的驅動程序:
sudo apt-get install nvidia-340
6.安裝 CUDA UVM Module
sudo apt-get install nvidia-340-uvm
7.重新開機
B.安裝CUDA Toolkit工具
1.首先下載對應版本的CUDA,這裏我們以run方式爲例:
https://developer.nvidia.com/cuda-downloads
2.CUDA下載完成之後,到對應的文件夾下,賦予cuda_xxxx.run可執行的權限;這裏假設CUDA下載到/home/user/download下:
cd /home/user/download/
chmod 755 ./cuda_xxxx.run
3.將剛剛下載的cuda_xxxx.run解壓爲三個文件:
cuda-xxxx.run(CUDA Toolkit)
NVIDIA-xxxx.run(nVIDIA Driver)
cuda-samplesxxxx.run(CUDA Samples)
cd /home/user/
mkdir ./temp
./cuda_xxxx.run --extract=/home/user/temp
cd /home/user/temp
4.安裝CUDA Toolkit
sudo ./cuda-xxxx.run
5.安裝好CUDA Toolkit之後,設定系統環境變量路徑;修改.bashrc
vim ~/.bashrc
在檔案最後兩行加上:
export PATH=/usr/local/cuda-xxx/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xxx/lib64:$LD_LIBRARY_PATH
6.重載環境變量
source ~/.bashrc
C.安裝CUDA Samples進行CUDA是否安裝成功的檢測
1.安裝CUDA Samples
sudo ./cuda-xxxx.run
2.完成安裝之後,編譯sample,檢測CUDA是否正常運行:
sudo su
cd /usr/local/cuda-xxx/sample/1_Utilities/deviceQuery
make
./deviceQuery