Loading

Ubuntu18.04--install NVIDIA-Driver CUDA cuDNN

系统环境:Ubuntu18.04
显卡型号:NVIDIA GeForce RTX 2080 SUPER
架构信息:Turing(CUDA 10 and later),SM75 or SM_75, compute_75 –
Driver Version:470.129.06
CUDA Version:11.4
CUDNN_VERSION:8204 (8.2.4)
Host compiler version : GCC 7.5.0

驱动安装

两种安装方式:

  • Ubuntu更新软安装 - 打开Software & Updates软件 -> Additional Drivers -> 等待搜索完毕 -> 选择NVIDIA Corporation下面的'Using NVIDIA Server Driver metapackage from nvidia-driver-470-server(proprietary)' -> 点击Apply Changes
  • NVIDIA官网驱动搜索下载安装,注意驱动版本与显卡的兼容性

CUDA安装

terminal输入nvidia-smi查看最高支持的CUDA版本号

我的是11.4(即显示信息表格第一行的最右边哪个CUDA Version:11.4,表示此驱动最高支持的CUDA版本)

NVIDIA-CUDA下载页面查看需要下载的CUDA版本,命令行下载安装

# 下载
wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
# 安装,回车之后需耐心等待才会出来安装界面,空格取消勾选驱动安装,其他全选
sudo sh cuda_11.4.4_470.82.01_linux.run

添加环境变量-参考
vim ~/.bashrc 文件末尾添加:

# Add the related bin and lib for cuda toolkit
export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"

然后source ~/.bashrc
如果使用了zsh则会报错
解决方法:exec bash 切换到bash再source ~/.bashrc
但是此种方法只能在bash上使用,切换回zsh依然找不到,说明zsh的配置文件.zshrc与bash的配置文件.bashrc是相对独立的需要额外在.zshrc里面配置
vim ~/.zshrc 文件末尾添加相同信息,再source ~/.zshrc

检查CUDA是否安装成功

1 - nvcc --version
2 - 编译测试设备 deviceQuery(路径仅供参考)

cd /usr/local/cuda-11.4/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
# 结果为:Result = PASS代表通过无问题

3 - 编译并测试带宽 bandwidthTest

cd ../bandwidthTest
sudo make
./bandwidthTest
# 结果为:Result = PASS代表通过无问题

cuDNN安装

cuDNN Archive 官网 查看相关版本,下载安装包安装

显卡驱动和cuda版本存在一一对应的版本范围,nvidia-smi查看驱动版本470兼容的最高cuda版本为11.4,所以下载cuDNN v8.2.4 for CUDA 11.4即可,下载三个文件:
cuDNN Runtime Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Developer Library for Ubuntu18.04 x86_64 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 x86_64 (Deb)
安装命令:
sudo dpkg -i libcudnn8_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda11.4_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda11.4_amd64.deb

检查cuDNN安装成功

cp -r /usr/src/cudnn_samples_v8 ~
cd ~/cudnn_samples_v8
cd mnistCUDNN
make clean
make
./mnistCUDNN
# 结果为Test passed!代表cuDNN安装成功

编译mnistCUDNN时出错:fatal error: FreeImage.h: No such file or directory
sudo apt-get install libfreeimage3 libfreeimage-dev

相关参考链接:

posted @ 2022-07-31 15:30  MrSu  阅读(422)  评论(0编辑  收藏  举报