Ubuntu18.04更新/安装cuda v10.2&cuDNN v7.6.5

1. 下载cuda10.2和cuDNN v7.6.5

1.1 下载cuda

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

1.2 下载cuDNN

从Nvidia官网登陆后,下载cuDNN

2. (可选)卸载旧驱动

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean

3. (可选)删除旧版本cuda

cd /usr/local/cuda-10.1/bin
sudo ./cuda-uninstaller

4. (可选)手动安装Nvidia显卡驱动

插曲:我在卸载完驱动和旧版本cuda后,运行cuda_10.2.89_440.33.01_linux.run文件时选择同时安装自带的捆绑nvidia显卡驱动,但是不知道什么原因驱动安装失败,并且好像捆绑的驱动版本和我刚卸载的版本是一样的,只好手动重装驱动。

4.1 查看推荐的NVIDIA显卡驱动版本

ubuntu-drivers devices

4.2 根据4.1结果,安装NVIDIA驱动

sudo apt install nvidia-driver-XXX

注:XXX为4.1结果中显示的具体版本号

5 安装cuda&cuDNN

5.1 安装cuda v10.2

注意,因为已经手动安装适合版本的NVIDIA显卡驱动,因此在安装过程中不再勾选安装程序内捆绑安装的显卡驱动

sudo sh cuda_10.2.89_440.33.01_linux.run

安装完成后,根据安装程序提示,在.bashrc文件中添加以下两行

sudo vim ~/.bashrc
PATH="/usr/local/cuda-10.2/bin:$PATH"
LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64"

5.2 安装cnDNN v7.6.5

解压下载好的cnDNN文件

tar -zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

安装cnDNN

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

你可能感兴趣的:(环境搭建,cuda,ubuntu)