在ubuntu 22.04.4安装cuda和cudnn

Ubuntu系统中,使用nvidia-smi命令可以看到当前GPU信息,在右上角可以看到CUDA Version,意思是最大支持的CUDA版本号。

在ubuntu 22.04.4安装cuda和cudnn

拟安装:
cuda 11.8
cuDNN v8.9.0 (for CUDA 11.x)

安装CUDA

打开官网:https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA 11.8,如下图选择“Lunix”及对应的架构、系统、版本和安装形式:

在ubuntu 22.04.4安装cuda和cudnn选择runfile(local)形式,网站会给我们安装的命令:

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

依次在终端中输入这两条命令!

需要注意的事项有下面几个:

一、由于我们之前以及安装过驱动了,这次不要再安装驱动了。

在Driver那里按空格键去掉勾选!

二、选项最后面的核心也不要去勾选(它默认是非勾选状态的)

三、安装过程中需要手动输入“accept”接受;

配置环境变量

打开终端,配置环境变量:

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

输入:wq退出。

打开终端输入:

source ~/.bashrc

即可完成环境变量的配置!

为了验证有没有安装成功,可以在终端输入nvcc -V即可查看CUDA版本:

在ubuntu 22.04.4安装cuda和cudnn

安装cudnn

下载安装文件

打开并登陆 https://developer.nvidia.com/rdp/cudnn-archive
选择下载 cuDNN v8.9.0 (for CUDA 11.x),
再选择 Local Installer for Ubuntu22.04 x86_64 (Deb),
就开始下载了。下载完后,建议把deb文件移动到home主目录

这里需要注册账户并且登录才能下载!

文件下载好之后剪切到主文件夹!

在ubuntu 22.04.4安装cuda和cudnn

安装

然后在终端分别输入如下的命令进行安装!

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.0.131/cudnn-local-2063C34E-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install libcudnn8=8.9.0.131-1+cuda11.8
sudo apt-get install libcudnn8-dev=8.9.0.131-1+cuda11.8
sudo apt-get install libcudnn8-samples=8.9.0.131-1+cuda11.8
sudo apt-get install libfreeimage3 libfreeimage-dev

验证是否安装成功

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

在ubuntu 22.04.4安装cuda和cudnn

当看到“Test passed!”就意味着安装成功了!

原创文章,作者:朋远方,如若转载,请注明出处:https://caovan.com/zaiubuntu-22044anzhuangcudahecudnn/.html

(0)
打赏 微信扫一扫 微信扫一扫
朋远方的头像朋远方
上一篇 2024年6月25日 上午7:55
下一篇 2024年6月28日 下午3:42

相关推荐

发表回复

登录后才能评论