因为电脑漏洞的缘故,前几天将Ubuntu18.04换成了20.04,无奈要重装PyTorch等环境,只有CUDA11.0以上版本才支持Ubuntu20.04,所以安装了CUDA11.0、cuDNN8.0.5及PyTorch1.7.0。这里记录下过程,以备后用。
1、首先在Nvidia官网下载适合自己机器的CUDA版本,官网下载,本次使用 runfile 的方式安装(用deb格式安装的话,会在安装过程中替换掉已经安装好的显卡驱动),如图所示:
在终端输入: 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,如下图,
用一个简单的方法移除已安装的Nvidia驱动,在Software & Updates里面找到Additional Drivers,选择Ubuntu系统自带的驱动,Apply changes后,重启,如下图
重新输入: sudo sh cuda_11.0.3_450.51.06_linux.run
下面界面中输入 accept 回车,
下面选择 Install,
安装完成,显示如下图,
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}}