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

因为电脑漏洞的缘故,前几天将Ubuntu18.04换成了20.04,无奈要重装PyTorch等环境,只有CUDA11.0以上版本才支持Ubuntu20.04,所以安装了CUDA11.0、cuDNN8.0.5及PyTorch1.7.0。这里记录下过程,以备后用。

1、首先在Nvidia官网下载适合自己机器的CUDA版本,官网下载,本次使用 runfile 的方式安装(用deb格式安装的话,会在安装过程中替换掉已经安装好的显卡驱动),如图所示:

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

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

在终端输入:  wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run

下载完成后输入: sudo sh cuda_11.0.3_450.51.06_linux.run 

如果电脑没有安装Nvidia显卡驱动,安装会顺利进行;若已经安装了Nvidia驱动,会提醒移除,此时我们先Abort,如下图,

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

用一个简单的方法移除已安装的Nvidia驱动,在Software & Updates里面找到Additional Drivers,选择Ubuntu系统自带的驱动,Apply changes后,重启,如下图

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

重新输入: sudo sh cuda_11.0.3_450.51.06_linux.run

下面界面中输入 accept  回车,

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

下面选择 Install

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

安装完成,显示如下图,

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

2、根据提示的信息,我们需要配置环境变量,打开 bashrc 文件,

输入: sudo gedit ~/.bashrc ,然后在文件末尾添加下面3行(第1行是注释),保存

# CUDA Soft Link export PATH=http://www.likecs.com/usr/local/cuda-11.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=http://www.likecs.com/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zwdxzj.html