ubuntu安装cuda和cudnn

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

满满满满红

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

阅读量1.8k

点赞数 2

文章标签： ubuntu linux python

本文链接：https://blog.csdn.net/weixin_45115648/article/details/130881358

版权

该文详细介绍了如何安装CUDA11.4，包括检查当前CUDA版本，通过runfile安装，设置环境变量，以及验证安装是否成功。此外，还阐述了cudnn的下载、解压、复制到CUDA库路径并进行验证的过程。

摘要由CSDN通过智能技术生成

一、查看驱动及cuda版本

nvidia-smi

在这里插入图片描述
显示cuda最高可以安装11.4版本

cuda下载链接
选择以前版本安装，选择我们需要的cuda11.4下载
在这里插入图片描述
这里强调一下，建议选择runfile

键盘上下键选中Driver，按空格去掉中括号里的X，例如上图所示，因为驱动已经安装好了，不需要在安装了Driver是驱动

这样就代表安装成功

安装bin文件lib要被系统识别到，需要在用户对应的环境变量设置里添加，bashrc

vim ~/.bashrc

export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-11.4/targets/x86_64-linux/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

以上注意的点
11.4为我安装的版本，具体要去根据每个人的版本去核对
第三个环境变量，实际可能并不需要安装。主要为前俩条

nvcc -V

在这里插入图片描述

cd /usr/local/cuda-11.4/samples/1_Utilities/deviceQuery/

sudo make

./deviceQuery

在这里插入图片描述
显示Result = PASS 即可

cudnn下载
cudnn下载需要账号，可以自己注册。
在这里插入图片描述
选择符合自己cuda版本的tar包，下载下来，进行解压

tar -xvf cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz

cd cudnn-linux-x86_64-8.9.1.23_cuda11-archive

sudo cp include/cu*.h /usr/local/cuda-11.4/include

sudo cp lib/libcudnn* /usr/local/cuda-11.4/lib64

sudo chmod a+r /usr/local/cuda-11.4/include/cu*.h

sudo chmod a+r /usr/local/cuda-11.4/lib64/libcudnn*

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

在这里插入图片描述
这样即可

关注