基本環境及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/