安装CUDA前需要查看设备是否安装好NVIDIA驱动,具体步骤可以查看我的上一篇安装教程。
云服务器Ubuntu20.04配置NVIDIA驱动+CUDA+cudnn+Anaconda
由于使用runfile格式安装的CUDA和cudnn总是报错,尝试各种解决方案均没有作用,无奈之下尝试使用deb格式重新安装,发现问题解决。
CUDA安装
进入CUDA官网:CUDA Toolkit Archive
选择好自己设备的信息,并选择deb格式安装
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.3.0/local_installers/cuda-repo-ubuntu2004-12-3-local_12.3.0-545.23.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-3-local_12.3.0-545.23.06-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-3-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-3
按照Base Installer提供的代码进行安装即可。(这是我需要安装的CUDA版本,其他版本按照流程进行安装即可)
添加系统路径
进入bashrc文件
sudo nano ~/.bashrc
在最后添加
#将下面两行插入到文件末尾
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
Ctrl+X,按Y,按回车
source ~/.bashrc #更新文件
验证CUDA安装
nvcc -V
图为我用上一篇文章的,正常应该显示CUDA12.3 。
安装cudnn
进入cudnn的官网:cuDNN Archive
选择CUDA的对应的版本和系统对应的版本下载
我选择Ubuntu20.04 x86_64(Deb)下载
下载后会得到一个压缩包,将其解压,并进入解压后的文件夹内
输入下面命令
sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb
导入CUDA GPG key
sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
刷新数据
sudo apt-get update
分别安装三个库
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
x.x.x是对应下载的cudnn版本信息,X.Y是 cuda的大版本和小版本信息
如果找不到需要填写的内容,可以进入/var/cuda-repo-ubuntu2004-12-3-local/路径下查看
ls -flt
会出现对应的deb文件,复制版本信息到上面三条命令进行安装。
安装完成后,查找文件路径
whereis cudnn_version.h
使用以下任意一条命令验证安装是否成功:
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 5 MAJOR
将路径替换为你查找到的路径,运行成功会出现如下内容:
Anaconda安装可以参考我的另一篇文章: