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.04ubuntu.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-510.47.03.run --uninstall
将nouveau添加到黑名单，这里需要sudo权限：
sudo gedit /etc/modprobe.d/blacklist.conf
最后一行添加：
blacklist nouveau
之后，手动下载NVIDIA驱动程序：
下载NVIDIA-Linux-x86_64-510.47.03.run
执行命令：sh NVIDIA-Linux-x86_64-510.47.03.run
安装成功后
验证驱动
执行命令：nvidia-smi
显示显卡信息
3.2 安装CUDA
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
第四步，安装Pytorch
下载:torch-1.9.0+cu102-cp39-cp39-linux_x86_64.whl
torchaudio-0.9.0-cp39-cp39-linux_x86_64.whl
torchvision-0.10.0+cu111-cp39-cp39-linux_x86_64.whl
执行命令：
pip install torch-1.9.0+cu102-cp39-cp39-linux_x86_64.whl
pip install torchaudio-0.9.0-cp39-cp39-linux_x86_64.whl
pip install torchvision-0.10.0+cu111-cp39-cp39-linux_x86_64.whl
第五步，验证环境
>>> import torch
>>> torch.cuda.is_available()
True
结果是true就ok了。如下图
