Ubuntu16.04安裝Cuda

基本環境及cuda(包括NVIDIA驅動以及cudnn)

1、製作U啓動,安裝好系統

2、sudo apt update

3、sudo apt upgrade #(此操作更新列表裏可更新的軟件)

4、sudo apt-get install build-essential #(安裝一些基礎的環境)

方案一:(該方法會把nVidia驅動同時裝上)

5、下載cuda8.0的DEB local安裝包

6、sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb

7、sudo apt update

8、sudo apt install cuda

9、reboot

方案二:(該方法會讓你選擇是否需要安裝顯卡驅動)

5、下載cuda8.0的runfile local(即.run)安裝包

6、sudo sh cuda_9.0.176_384.81_linux.run

7、按照提示和自己的需求安裝(工具、驅動等)

8、9、reboot

10、往~/.bashrc中添加cuda環境變量

export CUDA_HOME=/usr/local/cuda-9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

11、查看環境中cuda版本

  nvcc -V

12、驗證cuda安裝正確性:

cd到cuda測試例子目錄(eg:/usr/local/cuda-9.0/samples/1_Utilities/deviceQuery)

sudo make
./deviceQuery

看到輸出中有:
Result = PASS
恭喜你,cuda安裝成功

13、解壓cudnn包

14、放在指定位置例如:(或/usr/lib/x86_64-linux-gnu)

sudo cp include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp lib64/* /usr/local/cuda-9.0/lib64/
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章