一.安裝cuda11.0
cuda下載
此處的cuda下載請根據自己的系統信息下載,否則使用會有問題。
安裝說明 :
wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run
1.啟動安裝程序,一直按空格到最后,輸入accept接受條款
2.不安裝nvidia圖像驅動,之前已經安裝過了
3.安裝完成后會有安裝的日志記錄輸出, cat /var/log/cuda-installer.log,查看無報錯一般就是安裝成功了,
4.配置環境變量
vim ~/.bashrc
加入路徑:
export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64/
刷新配置:source ~/.bashrc
5.cuda驗證
nvcc -V
5.測試
首先安裝g++
apt-get install g++
#編譯並測試設備 deviceQuery:
cd /usr/local/cuda-11.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
#編譯並測試帶寬 bandwidthTest:
cd ../bandwidthTest
sudo make
./bandwidthTest
如果這兩個測試的最后結果都是Result = PASS,說明CUDA安裝成功啦
二.安裝cuDNN
查找nvcc -V對應的版本下載
1.安裝
解壓到cuda文件夾內 運行以下命令,進行安裝
cd cuda/include/
sudo cp ./include/cudnn.h /usr/local/cuda/include #執行即可,不需要做更改
sudo cp ./lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
2.建立軟連接
可以到下/usr/local/cuda/lib64/看一下自己的文件
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.8
sudo ln -s libcudnn.so.8.0.5 libcudnn.so.8
sudo ln -s libcudnn.so.8 libcudnn.so
保存后,創建鏈接文件
sudo vim /etc/ld.so.conf.d/cuda.conf
按a進入插入模式,增加下面一行
/usr/local/cuda/lib64
按esc退出插入模式,按:wq保存退出
最后在終端輸入
sudo ldconfig
后續Anaconda3等的安裝請參考此處:https://www.cnblogs.com/KdeS/p/14741567.html
文章參考地址:https://blog.csdn.net/pursuit_zhangyu/article/details/79362128