Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过

Ubuntu 20.04安装cuda-toolkit-11-4以及cuDNN

  • 一、配置好显卡驱动
  • 二、cuda-toolkit-11-4
    • 1、观察自己显卡最高支持的版本
    • 2、下载CUDA
  • 三、安装cuDNN

ubuntu截图方法:
加个Ctrl键,截图保存到剪贴板:

Ctrl+PrtSc:截取整个屏幕保存到剪贴板(类似windows全屏幕截图)
Ctrl+Alt+PrtSc:截取当前窗口保存到剪贴板(类似windows窗口截图)
Ctrl+Shift+PrtSc:任意选取矩形区域截图保存到剪贴板(类似windows矩形截图)

一、配置好显卡驱动

按照我之前的这篇博文进行配置:
https://blog.csdn.net/sjjsbsbbs/article/details/123791491

二、cuda-toolkit-11-4

1、观察自己显卡最高支持的版本

打开终端,输入:

nvidia-smi

观察右上角最高支持的版本,后续安装不高于这个版本,我的显卡(3050显卡)最高支持11.6。
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第1张图片

2、下载CUDA

可直接去官网下载所需版本:
https://developer.nvidia.com/cuda-toolkit-archive
如果你的显卡支持11.4,也可直接运行下述语句:
下载CUDA:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt-get update

再安装cuda-toolkit

apt-get install -y cuda-toolkit-11-4

测试一下cuda-samples

cd /usr/local/cuda-11.4/samples/4_Finance/BlackScholes
make BlackScholes
./BlackScholes

如果出现下面的结果,代表成功了:
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第2张图片
输入nvcc -V,查看版本信息,如果出现11.4或你自己下载的就说明成功了,如果出现的版本不对,那就是nvcc指向不对,需要进行修改,指向自己的CUDA。
执行

cd /usr/bin
vi /usr/bin/nvcc

没有vi就先下载,sudo apt-install vim
输入i进行修改,中间的cuda-11.4改成自己的版本就行,保存完成后esc退出,最后执行:wq!强制保存,
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第3张图片
完成后再次输入:

nvcc -V

如果是自己下载的对应版本说明成功了。

三、安装cuDNN

先去官网进行下载:
https://developer.nvidia.cn/rdp/cudnn-archive
需要找到和CUDA对应的版本,要下载三个deb文件:
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第4张图片
点击对应的xxx for CUDA11.4,然后会出现如下的列表:
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第5张图片
下载这三个:
Ubuntu 20.04配置cuda-toolkit-11-4以及cuDNN一遍过_第6张图片
下载好后会保存在本地的Dowload文件夹,cd到这个文件夹下,然后依次执行下列代码,输入前面的,按下tab键自动补全:

sudo dpkg -i libcudnn8_+tab键
sudo dpkg -i libcudnn8-dev+tab键
sudo dpkg -i libcudnn8-samples+tab键

接下来执行下述语句:
#复制cuDNN samples到home目录下

cp -r /usr/src/cudnn_samples_v8 /$HOME

#进入home目录

cd $HOME/cudnn_samples_v8/mnistCUDNN/

#编译mnistCUDNN

sudo make clean
sudo make

#运行mnistCUDNN

sudo ./mnistCUDNN

执行

sudo make

#运行mnistCUDNN

sudo ./mnistCUDNN

显示test passed说明安装成功。
但是执行:

sudo make

时候,可能会报错:

1 | #include "FreeImage.h"
  |          ^~~~~~~~~~~~~

此时安装:

sudo apt-get install libfreeimage3 libfreeimage-dev

如果安装不成功,可能是源的问题,则需要配置国内源了。
参考下列博客:
https://blog.csdn.net/weixin_45712255/article/details/108017677
最后重新执行一遍
执行

sudo make

#运行mnistCUDNN

./mnistCUDNN

如果显示test passed说明安装成功。
至此,安装完成

你可能感兴趣的:(python,安装教程,python)