【linux】服务器Ubuntu20.04安装cuda11.8教程
文章目录
-
- 【linux】服务器Ubuntu20.04安装cuda11.8教程
- 到官网找到对应版本下载链接
- 终端操作
- cudnn安装
-
- 到官网下载
- 下载后解压
- 进入解压后的目录:
- [将头文件复制到 /usr/local/cuda/include/ 目录:](#将头文件复制到 /usr/local/cuda/include/ 目录:)
- [将库文件复制到 /usr/local/cuda/lib64/ 目录:](#将库文件复制到 /usr/local/cuda/lib64/ 目录:)
- 修改文件权限,以确保所有用户都可以读取这些文件:
- 验证cuDNN安装是否成功:
- 设置环境变量
到官网找到对应版本下载链接
这里我们选择以往版本
终端操作
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