【linux】服务器Ubuntu20.04安装cuda11.8教程

【linux】服务器Ubuntu20.04安装cuda11.8教程

文章目录

到官网找到对应版本下载链接

这里我们选择以往版本

终端操作

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

cudnn安装

到官网下载

下载后解压

tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar

进入解压后的目录：

cd cudnn-linux-x86_64-8.9.3.28_cuda11-archive

将头文件复制到 /usr/local/cuda/include/ 目录：

sudo cp include/cudnn*.h /usr/local/cuda-11.8/include/

将库文件复制到 /usr/local/cuda/lib64/ 目录：

sudo cp lib/libcudnn* /usr/local/cuda-11.8/lib64/

修改文件权限，以确保所有用户都可以读取这些文件：

sudo chmod a+r /usr/local/cuda-11.8/include/cudnn*.h /usr/local/cuda-11.8/lib64/libcudnn*

验证cuDNN安装是否成功：

cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

设置环境变量

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH