ubuntu下CUDA以及cudnn安装

最新推荐文章于 2024-05-04 15:05:24 发布

Qiyao .C

最新推荐文章于 2024-05-04 15:05:24 发布

阅读量1.4k

点赞数 2

分类专栏：视觉内容学习文章标签： ubuntu linux 机器学习深度学习

本文链接：https://blog.csdn.net/qq_19449259/article/details/129925442

版权

5 篇文章 3 订阅

订阅专栏

下载库

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

本文以安装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

nvcc -V

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

此处假设已经下载好了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

关注