1.安裝cuda11.1
- 在nvidia官網(wǎng)下載與445.38驅(qū)動版本對應的cuda版本11.1.1椒丧,地址為https://developer.nvidia.com/cuda-toolkit-archive晴股。建議采用runfile(local)方式下載安裝。安裝cuda時鸳惯,不要選中安裝driver兵志,其余都選中安裝.
wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.runsudo
sh cuda_11.1.1_455.32.00_linux.run
- 打開主目錄下的 .bashrc文件添加如下路徑杂曲,.bashrc文件在/home下冗美,如果沒有找到,則按Ctrl+H鍵顯示隱藏文件掸屡。將以下命令添加至文件末尾封寞,保存退出。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
source ~/.bashrc
注:上述命令中的cuda也可以改為cuda11.1仅财,因為在系統(tǒng)目錄中有cuda和cuda11.1兩個文件夾狈究,其中cuda文件夾是cuda11.1的鏈接,二者內(nèi)容相同盏求。
- 檢查:nvcc -V抖锥,查看cuda是否正確安裝
2.安裝cudnn8.0.4
- 在NVIDIA官網(wǎng)下載與cuda11.1版本對應的cudnn8.0.4亿眠,下載地址為:https://developer.nvidia.com/rdp/cudnn-archive。建議下載cuDNN Library for Linux (x86_64)安裝包磅废。
- 安裝cudnn v8.0.4
sudo cp cudnn-11.1-linux-x64-v8.0.4.30.solitairetheme8 cudnn-11.1-linux-x64-v8.0.4.30.tgz
tar -xzvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
注:此處必須使用sudo cp cuda/include/* /usr/local/cuda/include/缕探,而不是sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
https://developer.nvidia.com/tensorrt-download
安裝TensorRT
https://blog.csdn.net/Williamcsj/article/details/123514435
卸載cuda
(108條消息) Ubuntu18.04安裝CUDA10.2 + cuDNN v7.6.5_堅持健身的程序員的博客-CSDN博客