Pytorch1.9 GPU CUDA111安装教程

1 NVIDIA-3090Ti

安装好NVIDIA显卡驱动后，使用nvidia-smi 语句进行查询目前本机的CUDA型号。
安装Anaconda后，建立虚拟环境，使用conda create -n torch1.9 python=3.8 创建一个名字为torch1.9的虚拟环境。
激活虚拟环境source activate torch1.9
Pytorch官网地址

2 安装GPU版本

官网提示torch1.9GPU支持CUDA10.2和CUDA11.1，因本机显卡为NVIDIA-3090Ti所以，算力较强，以CUDA11.1为例安装。基于刚创建的虚拟环境torch1.9下，使用如下命令安装：

`pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html`

安装结束后，在虚拟环境的终端输入python，之后输入以下代码：

import torch
torch.cuda.is_available()

返回True，表示环境可用。

上述为pytorch gpu版本的安装过程，等程序运行时，需要哪个包就pip insatall 即可。如果网络不好可以先去下载相应版本的安装包，再使用pip install 安装包名称 即可。下载地址为：

https://download.pytorch.org/whl/torch_stable.html