天天看點

64位Ubuntu14.04+GTX960安裝CUDA8.0及cudnn

0.檢查gpu,系統版本和核心是否支援,一般沒問題。

  • 部分計算機需要在BIOS中關閉secure boot
  • 如果出現黑屏問題,請看這裡

1.gcc和g++調整:安裝cuda對應的gcc和g++并進行軟連結,本教程cuda對應4.8

sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install -y gcc-4.8
sudo apt-get install -y g++-4.8
sudo su
cd ../../usr/bin
ln -s /usr/bin/g++-4.8 /usr/bin/g++ -f
ln -s /usr/bin/gcc-4.8 /usr/bin/gcc -f
           

2.安裝32位相容包。因為安裝二進制安裝包時,提示是否安裝32bit相容包,是以,最好安裝上32相容包,以後作業系統上安裝32bit軟體也友善。

sudo apt-get install -y lib32z1
           

3.禁用nouveau

lsmod | grep nouveau   檢視nouveau是否使用,無輸出則意味着被禁用或沒有
有則執行sudo vim /etc/modprobe.d/blacklist-nouveau.conf, 輸入
blacklist nouveau
options nouveau modeset=0
儲存退出
在終端執行sudo update-initramfs -u
重新開機電腦
           

4.安裝CUDA

sudo service lightdm stop 關閉圖形界面,按ctrl+alt+F1進入tty1并登入
進入run檔案所在目錄
chmod 777 cuda_8.0.61_375.26_linux.run
sudo ./cuda_8.0.61_375.26_linux.run 需要時間讀取許可,按q退出閱讀許可
依次接受許可,安裝driver:yes,安裝toolkit和samples:yes,中間選擇及安裝路徑直接Enter選擇預設
sudo service lightdm start 打開圖形界面
執行vim  ~/.bashrc,在檔案最後加入
export PATH=/usr/local/cuda/bin:$PATH  
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
執行source ~/.bashrc使配置的環境變量生效 
nvidia-smi 出現GPU相關資訊表示驅動安裝成功
執行nvcc --version檢視cuda版本資訊
           

5.測試cuda的Samples

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery  
sudo make  
sudo ./deviceQuery
保證編譯沒有error,執行deviceQuery輸出資訊的最後類似以下:
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0, CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GTX 960
Result = PASS
           

6.cudnn安裝

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz  
sudo cp cuda/include/cudnn.h /usr/local/cuda/include  
sudo cp cuda/lib64/*.* /usr/local/cuda/lib64 -a
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
           

7.解除安裝

sudo nvidia-uninstall 或者
sudo ./cuda_8.0.61_375.26_linux.run --uninstall 或者
sudo apt-get remove --purge nvidia-* 解除安裝驅動
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl 解除安裝cuda
           

繼續閱讀