Ubuntu 20.04-Pytorch-GPU系统搭建指南

本人硬件环境:

CPU:Intel Core i7 6700

GPU:NVIDIA GTX 1060 6G

内存:SAMSUNG DDR4 32GB

硬盘:双SSD 256G,双系统(windows,ubuntu)

第一步,安装Ubuntu.

1.1 准备工作

1.1.1 下载 Ubuntu 镜像

打开ubuntu 20.04的下载地址,进入页面后点击右边的Download按钮开始下载。

Ubuntu 20.04-Pytorch-GPU系统搭建指南

Ubuntu 20.04​ubuntu.com/download/desktop

你会得到一个Ubuntu ISO文件。

制作U盘启动盘然后安装即可。

第二步,配置Python环境.

下载Anaconda3-2021.11-Linux-x86_64.sh

执行命令:bash Anaconda3-2021.11-Linux-x86_64.sh

安装晚后restart

第三步,安装NVIDIA驱动,配置CUDA+

首先,建议你需要满足这两个条件,虽然没有任何证据证明与成功安装的相关性,但是这样做的确会减少报错:

  • BIOS务必关闭Secure Boot.
  • Python3版本应不低于3.5.

如果你按照上面的操作步骤,你只需要在BIOS里关闭Secure Boot。

3.1卸载已有的驱动程序,禁用nouveau

#卸载之前的所有驱动
#apt-get安装的原有驱动:
sudo apt-get remove --purge nvidia*
 
#runfile安装的原有驱动:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-495.46.run --uninstall
将nouveau添加到黑名单,这里需要sudo权限:

sudo gedit /etc/modprobe.d/blacklist.conf
最后一行添加:

blacklist nouveau

之后,手动下载NVIDIA驱动程序:

Download Drivers​www.nvidia.com/Download/index.aspx?lang=en-us正在上传…重新上传取消

下载NVIDIA-Linux-x86_64-495.46.run

执行命令:sh NVIDIA-Linux-x86_64-495.46.run

安装成功后

验证驱动
执行命令:nvidia-smi

显示显卡信息

NVIDIA-SMI 495.46       Driver Version: 495.46       CUDA Version: 11.5  

3.2 安装CUDA

下载:cuda-repo-ubuntu2004-11-5-local_11.5.0-495.29.05-1_amd64.deb

执行命令:

sudo dpkg -i cuda-repo-ubuntu2004-11-5-local_11.5.0-495.29.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-5-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

第四步,安装Pytorch

执行命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

第五步,验证环境

上一篇:Linux进程在后台可靠运行


下一篇:3、蜂鸣器实验