0
点赞
收藏
分享

微信扫一扫

ubuntu24安装cuda和cudnn

hwwjian 2024-09-05 阅读 90
linux

一、安装cuda
确保显卡驱动正确安装

终端输入:
nvidia-smi

显示下面结果,说明显卡驱动安装正常,可以进行下一步
在这里插入图片描述1.去官网下载CUDA,需要注册账号下载

https://developer.nvidia.com/cuda-toolkit-archive在这里插入图片描述由于我们显卡支持12.2,因此为了保险起见,我们安装低一个版本的cudn,选择12.1安装,在上述界面中点击“CUDA Toolkit 12.1.1(July 2023), Versioned Online Documentation”,然后进去下面界面
在这里插入图片描述2. 然后执行下面命令进行下载安装

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run

sudo sh cuda_12.1.1_530.30.02_linux.run

进入以下界面
在这里插入图片描述点击Continue“”

在这里插入图片描述输入"accept"

在这里插入图片描述
在这里,按“空格”键取消Driver,因为前面已经安装过驱动,再次安装可能出现冲突。然后选择“Install”
3.等待安装完成之后,配置环境变量。

export PATH=$PATH:/usr/local/cuda-12.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.1
source ~/ .bashrc
  1. 上述完成之后,再输入
nvcc -V

输出下面信息,表示安装成功
在这里插入图片描述
二、安装cudnn
1.去官网下载对应的cudnn

https://developer.nvidia.com/rdp/cudnn-download

在这里插入图片描述
2.依次执行下面命令进行安装

wget https://developer.download.nvidia.com/compute/cudnn/9.3.0/local_installers/cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.debsudo 
cp /var/cudnn-local-repo-ubuntu2404-9.3.0/cudnn-*-keyring.gpg /usr/share/keyrings/(这句话在执行上面安装完成之后界面会显示,直接复制界面中提示的执行即可)
sudo apt-get updatesudo apt-get -y install cudnn
sudo apt-get -y install cudnn-cuda-12

3.为了确保下面拷贝的程序的位置准确,通过whereis命令查找一下cudnn的安装位置

whereis cudnn_version.h
输出
cudnn_version.h: /usr/include/cudnn_version.h

所以,cudnn的安装文件在/usr/include下面

4.执行文件复制

sudo cp usr/include/cudnn* /usr/local/cuda-12.1/include

5.执行下面一句话,测试是否安装cudnn成功

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

在这里插入图片描述同时可以使用

dpkg -l |grep cudnn
查看版本信息

在这里插入图片描述注意!安装完CUDA并通过nvcc -V命令测试完毕之后再次安装cudnn,可能由于复制文件过程中出现问题,导致安装完cudnn并测试成功后,再执行nvcc -V命令不显示,此时需要再重复步骤一即可,完成之后不需要重复步骤二即可完成安装任务。

举报

相关推荐

0 条评论