Ubuntu20.04安装Cuda、cuDNN

mmidge

已于 2024-03-22 11:44:02 修改

阅读量593

点赞数 6

文章标签：深度学习

于 2024-03-15 18:03:51 首次发布

本文链接：https://blog.csdn.net/zongmuyang/article/details/136745113

版权

自用在Ubuntu20.04安装cuda、cudnn

你好！这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Markdown编辑器, 可以仔细阅读这篇文章，了解一下Markdown的基本语法知识。

rm -r <文件名>

官网地址：https://developer.nvidia.com/cuda-downloads
找到之前发布的版本：在这里插入图片描述
选择匹配版本：

执行官网的安装指令

wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

sudo sh cuda_11.3.1_465.19.01_linux.run

安装细节见：Ubuntu 安装 CUDA11.3
接下来进行环境配置:

vim ~/.bashrc

翻到文件筐的最下面，光标移到最左边，点击o

输入

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

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

点击esc按键，输入“：wq”保存冒号wq

source ~/.bashrc
# 查看cuda或驱动是都安装成功
nvcc -V   #输出安装成功的版本
nvidia-smi

官网地址：https://developer.nvidia.com/cudnn
进入官网，选择指定的版本。安装完成后可在 /usr/local/ 下看到安装好的cudnn

下载之后进入压缩包所在文件夹，执行解压指令：

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

在路径中生成cudnn-linux-x86_64-8.5.0.96_cuda11-archive文件夹，然后复制 lib64 和 include 文件夹到 /usr/local/cuda-11.3，命令如下：

sudo cp cudnn-linux-x86_64-8.5.0.96_cuda11-archive/include/cudnn.h /usr/local/cuda-11.3/include

sudo cp cudnn-linux-x86_64-8.5.0.96_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.3/lib64

sudo chmod a+r /usr/local/cuda-11.3/include/cudnn.h /usr/local/cuda-11.3/lib64/libcudnn*

验证

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

import torch
print(torch.__version__)
print("gpu", torch.cuda.is_available())

关注