Ubuntu下CUDA的安装及配置(run方式)

一、参考资料

CUDA、CUDNN在windows下的安装及配置

Ubuntu 20.04安装CUDA 11.0、cuDNN 8.0.5、PyTorch 1.7.0

二、安装前的准备

参考博客： CUDA、CUDNN在windows下的安装及配置

补充：CUDA版本与gcc版本匹配

Ubuntu18.04安装cuda10.0
NVIDIA官网版本对齐

（1）查看显卡驱动是否安装

>>> nvidia-smi

如果显卡驱动没有安装，不需要单独安装，安装CUDA过程中勾选显卡驱动（默认已经选择）

如果显卡驱动已经安装，在安装CUDA过程中不勾选显卡驱动

（2）NVIDIA官网下载对应CUDA版本

CUDA下载地址：https://developer.nvidia.com/cuda-toolkit-archive

 （3）CUDNN的下载

官网下载地址：https://developer.nvidia.com/rdp/cudnn-download

没有账号需要注册账号，账号登录后，找到自己对应的CUDA版本，点击下载

三、安装过程

（1）安装CUDA

安装cuda
>>> sudo sh cuda_10.2.89_440.33.01_linux.run

配置环境变量
>>> sudo gedit  ~/.bashrc

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量
>>> source ~/.bashrc

查看cuda是否安装成功
>>> nvcc -V

由于已经安装了显卡驱动，不选Driver显卡驱动

查看cuda是否安装成功

（2）安装CUDNN

将下载的 cuDNN Library for Linux (x86_64) 解压，复制解压出来的文件复制到安装好的CUDA环境中

将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹
>>> sudo cp cuda/include/cudnn.h /usr/local/cuda/include

将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中
>>> sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

添加读取权限
>>> sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

（3）配置CUDA环境变量

# 配置环境变量
>>> sudo gedit  ~/.bashrc
 
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 
# 更新环境变量
>>> source ~/.bashrc
 
# 查看cuda是否安装成功
>>> nvcc -V

（4）测试是否安装成功

参考博客 显卡/cudn/cuDNN相关查询

四、可能存在的问题

1. 关于CUDA多版本切换的问题

参考博客 关于CUDA，cuDNN，TF，CUDA驱动版本兼容问题