linux下安裝cuda cudnn
參考博客:
https://blog.csdn.net/u014595019/article/details/53732015
https://blog.csdn.net/wanzhen4330/article/details/81699769
https://blog.csdn.net/auh_alias/article/details/80690556
一 cuda安裝
https://developer.nvidia.com/cuda-downloads
1下載run類型,安裝官網代碼提示:
accept
n(不要安裝driver,如果安了)取消x
2聲明環境變量
安裝完畢後
export PATH=/usr/local/cuda/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc 使環境變量生效
nvcc -V 查看安裝成功!
二 cudnn
https://developer.nvidia.com/rdp/cudnn-archive
1 安裝cudnn
解壓下載的文件,可以看到cuda文件夾,在當前目錄打開終端,執行如下命令:
tar -xvf cudnn-10.1-linux-x64-v7.6.3.tgz
sudo cp cuda/include/cudnn.h /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*
2 查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2