Ubuntu20.04安装cuda10.0、cudnn

安装cuda

gcc降级

因为Ubuntu20.04自带gcc-9，而cuda10.0支持gcc-4，因此要降级，但是gcc-4安装失败，尝试用gcc-7后面也安装成功了。

sudo apt-get install gcc-7 g++-7
# 设置gcc各版本的优先级
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
# 设置g++各版本的优先级
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
# 查看优先级
sudo update-alternatives --display gcc
sudo update-alternatives --display g++

安装过程

sudo sh cuda_10.0.130_410.48_linux.run

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

或

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}}

查看是否安装成功

nvcc --version
cat /usr/local/cuda/version.txt

安装cudnn

切换到下载文件所在目录，解压下载好的cuDNN压缩文件到当前目录：

tar zxvf ./cudnn-10.1-linux-x64-v7.6.5.32.tgz -C ./

将解压出的

• cuda/include/cudnn.h文件复制到/usr/local/cuda/include文件夹
• cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹

为上述文件添加读取和执行权限：（不必须）

sudo chmod 755 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*