ubuntu下CUDA以及cudnn安装

下载库

确定自己所需要的cuda版本以及对应的cudnn版本。
CUDA下载链接:
https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载链接:
https://developer.nvidia.cn/rdp/cudnn-download
*其中cudnn需要注册开发者账号进行下载

安装CUDA

本文以安装cuda11.7为例:
在这里插入图片描述
打开对应网站,选择自己所需版本,按照窗口内的指令输入即可,建议安装deb(local)
在这里插入图片描述
最后在自己的bash内添加cuda对应的path

gedit ~/.bashrc

添加如下命令

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

在这里插入图片描述
输入以下命令更新bash

source ~/.bashrc

检测cuda安装

nvcc -V

在这里插入图片描述
效果显著!!!

安装cudnn

此处假设已经下载好了cudnn的deb文件
在这里插入图片描述
根据官网安装文件指南:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
在这里插入图片描述
该部分来完成安装,笔者将按照本机上的cudnn库演示安装

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.5.0.96_1.0-1_amd64.deb
#后者需要修改为自己下载的cudnn库文件

在这里插入图片描述
复制该指令运行

sudo cp /var/cudnn-local-repo-ubuntu2004-8.5.0.96/cudnn-local-0579404E-keyring.gpg /usr/share/keyrings/

更新源,也就是把这个deb的内容添加到apt里

sudo apt-get update

此时我们能在apt内找到相应的库了
在这里插入图片描述
此时不用按照官方教程内的语句来逐字逐句输入了,直接安装这几个就行了;

sudo apt-get install libcudnn8 libcudnn8-dev libcudnn8-samples

优雅完成!!!!

猜你喜欢

转载自blog.csdn.net/qq_19449259/article/details/129925442
今日推荐