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*
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值