Ubuntu 20.04 CUDA&cuDNN安装方法全新整理

目录

1、安装显卡驱动

2、gcc降级

3、安装CUDA

4、安装cuDNN


1、安装显卡驱动

输入nvidia-smi命令查看支持的cuda版本

如果有驱动显示以下信息:

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第1张图片

如果没有,则会显示以下信息:

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第2张图片

如果无法查看,则说明尚未安装nvidia驱动,点击附加驱动,选择对应版本的驱动即可自动下载。

 

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第3张图片

点击应用更改,然后开安装驱动,安装完成后重启电脑,然后再使用nvidia-smi命令查看显卡驱动是否正常和支持的CUDA版本。

2、gcc降级

Ubuntu20.04自带的gcc版本为9.7.0,需要添加gcc7才可安装cuda10.2,输入命令安装gcc7

sudo apt-get install gcc-7 g++-7

查看gcc版本,可以看到目前系统中存在7和9两个版本

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第4张图片

使用update-alternatives进行版本切换,输入以下命令:

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50

此时输入sudo update-alternatives --config gcc命令查看gcc的默认版本,可以看到当前默认gcc版本为7,即切换成功。

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第5张图片

3、安装CUDA

从https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,选择18.04的版本。如下图:

 

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第6张图片

按照指令安装cuda 10.1版本。指令如下:

下载指令:

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
 
  
安装指令:
sudo sh cuda_10.1.243_418.87.00_linux.run

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第7张图片

 

点击continue

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第8张图片

输入accept

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第9张图片

cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install,等待安装完成。

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第10张图片

安装完成,如下图:

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第11张图片

 

配置环境变量

输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。

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

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

更新环境变量配置

 source ~/.bashrc

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第12张图片

至此cuda安装完成,输入nvcc -V命令查看cuda信息

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第13张图片

 

4、安装cuDNN

从https://developer.nvidia.com/cudnn下载相应版本的cudnn,需要登陆。选择与CUDA 10.1对应的版本(7.6.5),点开后选择 cuDNN Library for Linux,点击下载。

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第14张图片

选择与CUDA 10.1对应的版本(7.6.5),点开后选择 cuDNN Library for Linux,点击下载。

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第15张图片

更改usr/local/cuda/文件夹下面的include和lib64文件夹的权限。否则会出现如下问题:

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第16张图片

修改incude文件夹权限

 

Ubuntu 20.04 CUDA&cuDNN安装方法全新整理_第17张图片

常用的修改权限的指令

sudo chmod 600 ××× (只有所有者有读和写的权限)
sudo chmod 644 ××× (所有者有读和写的权限,组用户只有读的权限)
sudo chmod 700 ××× (只有所有者有读和写以及执行的权限)
sudo chmod 666 ××× (每个人都有读和写的权限)
sudo chmod 777 ××× (每个人都有读和写以及执行的权限)

将下载的文件解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹,将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:

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

至此cuDNN安装完成

 

你可能感兴趣的:(人工智能,linux)