Anaconda3安装PyTorch

1.Anaconda、CUDA及cuDNN
参考上篇博客：Anaconda、TensorFlow安装及环境配置
2.补充：NVIDIA GPU显卡驱动安装

#禁用桌面管理环境lightdm，如果没安装lightdm，Ubuntu默认使用的时gdm3
sudo service lightdm stop

卸载可能存在的旧版本NVIDIA驱动（不执行，重启后会存在问题）

sudo apt-get  --purge remove nvidia*

安装驱动需要的依赖

sudo apt-get update
sudo apt-get install dkms build-essential Linux-headers-generic

把nouveau驱动加入黑名单并禁用nouveau内核模块

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

在blacklist-nouveau.conf文末添加：

blacklist nouveau
options nouveau modeset=0

保存退出后执行：

sudo update-initramfs -u

重启：

reboot

3.下载并安装PyTorch
在PyTorch官网，找到下载的版本：

	conda install pytorch torchvision torchaudio cudatoolkit=10.0

或指定版本然后安装：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.0

4.验证是否安装成功（GPU版本）

import torch
torch.__version__	#输出torch版本
torch.cuda.is_available()	#返回True，则表示GPU可用