Ubuntu下安装CUDA和cuDNN

Ubuntu下安装CUDA和cuDNN

CUDA

1. 查看显卡是否支持CUDA

网址https://developer.nvidia.com/cuda-gpus。

2. CUDA版本选择

查看ubuntu版本是否对应要下载的CUDA，比如我选择的是11.7,对应CUDA TOOLKIT DOCUMENTION。

在system requirements中查看版本，在pre-installationo actions有所用的指令。

在这里选择对应CUDA版本，根据指示下载，我选择的是runfile方式。

但如果你和我一样是为了安装pytorch，保险起见看一下pytorch支持的cuda版本，因为我就下错了版本TT。

3. 安装

sudo sh [cuda_file_name]

后面会让你选择是否下载驱动，如果你已经下载好了，就不选择。如果没有下载，可以让它自动下载，或者你自助下载，参考这里：https://blog.csdn.net/weixin_48821657/article/details/127827774?spm=1001.2014.3001.5502

4. 配置环境变量

gedit ~/.bashrc打开文件

在末尾加上你所下载版本

export PATH=/usr/local/cuda-[version]/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-[version]/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量配置source ~/.bashrc

查看CUDA信息nvcc -V

cuDNN

下载链接https://developer.nvidia.com/rdp/cudnn-archive，选择与CUDA对应的版本，需要登陆。

参考NVIDIA CUDNN DOCUMENTATION安装。

解压tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz

将文件复制到CUDA对应路径下。

$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

完成啦～