600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > Ubuntu 20.04 安装NVIDIA显卡驱动+cuda 11.7+cudnn 8.4

Ubuntu 20.04 安装NVIDIA显卡驱动+cuda 11.7+cudnn 8.4

时间:2018-10-26 17:12:32

相关推荐

Ubuntu 20.04 安装NVIDIA显卡驱动+cuda 11.7+cudnn 8.4

Ubuntu 18.04 安装NVIDIA显卡驱动+cuda 10.2+cudnn

本机环境1 相关查询命令 一、Ubuntu 20.04 安装NVIDIA显卡驱动二、Ubuntu 20.04 安装cuda 11.71、安装CUDA 11.73、配置CUDA环境变量测试 三、cudnn8.4 安装

参考:

/p/59618999

/linhai1028/article/details/79445722/

Windows+Ubuntu从双系统安装到CUDA cuDNN docker 配置K21

/qq_45831128/article/details/127060475

/kunhe0512/article/details/125061911

本机环境

Ubuntu 20.04

英伟达 RTX 3050 Ti 显卡

1 相关查询命令

1)Ubuntu下查看Nvidia显卡的详细信息:

nvidia-smi

2)查看cuda 版本:

nvcc -V

3)查看cudnn版本:

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

4)在终端输入命令,实时查看GPU的使用情况:

watch -n 1 nvidia-smi

5)在Ubuntu下查询显卡型号

nvidia-smi -a |grep -i Product\ Name

一、Ubuntu 20.04 安装NVIDIA显卡驱动

下载NVIDIA-Linux-x86_64-515.57.run

这个版本驱动比较稳定。

二、Ubuntu 20.04 安装cuda 11.7

1、安装CUDA 11.7

cuda 下载地址:/zh-cn/cuda-downloads

第二步就是安装CUDA。这里我们安装CUDA最新的11.7版本。进入Nvidia cuda下载地址:

Nvidia cuda11.7 下载地址

我们这里选择的Ubuntu 20.04下的deb安装类型。

然后依次执行下述官方提供的几条命令:

wget https://developer./compute/cuda/repos/ubuntu/x86_64/cuda-ubuntu.pinsudo mv cuda-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer./compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu-11-7-local_11.7.0-515.43.04-1_amd64.debsudo dpkg -i cuda-repo-ubuntu-11-7-local_11.7.0-515.43.04-1_amd64.debsudo cp /var/cuda-repo-ubuntu-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/sudo apt-get updatesudo apt-get -y install cuda

3、配置CUDA环境变量

安装完成之后,加入到环境变量,可用gedit

gedit ~/.bashrc

在.bashrc末尾添加两行环境变量

export PATH=$PATH:/usr/local/cuda-11.7/binexport LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} #根据CUDA版本更换路径

最后,使环境变量生效

source ~/.bashrc

测试

nvcc -V

~$ nvcc -Vnvcc: NVIDIA (R) Cuda compiler driverCopyright (c) - NVIDIA CorporationBuilt on Tue_May__3_18:49:52_PDT_Cuda compilation tools, release 11.7, V11.7.64Build cuda_11.7.r11.7/compiler.31294372_0

三、cudnn8.4 安装

cudnn下载地址:/rdp/cudnn-archive

需要提前注册,找到对应版本,下载cuDNN Library for Linux压缩包即可

解压压缩包,并终端进入文件夹,执行以下

#进入解压目录cd cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive# 复制cudnn头文件sudo cp ./include/* /usr/local/cuda-11.7/include/ # 复制cudnn的库sudo cp ./lib/* /usr/local/cuda-11.7/lib64/ #给予权限sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*

(3)测试

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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。