文中使用的是cuda_12.1.0_530.30.02_linux.run
、cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb
1、Cuda安装
使用以下命令运行安装Cuda,随后跳出图中的选择。
sudo bash cuda_12.1.0_530.30.02_linux.run
因为安装过NVIDIA驱动了,所以将第一个Driver
的选项给取消
了,其他的保持不变,最后选Install
进行安装。
安装完成后再终端通过输入:
(这里注意-V是大写)
nvcc -V
显示图的内容,表示cuda安装成功。
2、Cudnn安装
通过使用以下命令解压cudnn文件。
sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb
得到图中信息,按照终端内的提示,进行sudo cp
的操作
2.1 进入/var/cudnn-local-repo-ubuntu2004-8.9.7.29
路径内,在这个路径下打开终端。(注意这个路径,是之前cp的文件,不同版本的cudnn 尾部内容会不一样)
在这个路径下,按顺序对libcudnn8
、libcudnn8-dev
、libcudnn8-samples
进行解压。结束后,cudnn就安装完成了