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声明环境变量
安装完毕后
sudo gedit ~/.bashrc
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