首先创建虚拟环境

conda create -n torch python=3.6(指定默认python为3.6)

激活torch

source activate torch

安装torch

Pytorch官方网站：https://pytorch.org/

安装时需要注意torch和torchvision的对应版本，如下所示
根据自己的系统和python已经cuda版本进行选择安装

官网命令安装

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio cudatoolkit=10.2 -c pytorch

注意这里安装时注意看安装的包是否包含cpuonly,如果包含的话，是无法调用GPU的，其次你如果需要用到自动混合精度训练包(amp)你需要安装PyTorch 1.6以上的版本，不然是不包含amp的

import torch.cuda.amp as amp

清华镜像安装（提高安装速度）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio cudatoolkit=10.2 -c pytorch

本地离线安装
镜像下载地址：

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

下载安装的镜像文件

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/torchvision-0.9.1-py36_cu102.tar.bz2

conda install --offline pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2
conda install  --offline torchvision-0.9.1-py36_cu102.tar.bz2

这里推荐官网命令安装和清华镜像安装（提高安装速度），离线安装可能会出现环境配置问题

安装好后，可以通过一下代码确认torch是否安装成功

import torch    
a = torch.Tensor([1.])   
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"

通过以下代码看安装的pytorch是GPU版本还是CPU版本，输出false则安装的是CPU版本的，true则是GPU版本

import torch
print(torch.cuda.is_available())

安装时如果遇到一下问题：

Fetching package metadata .......
CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://nanomirrors.tuna.tsinghua.edu.cn/anaconda/cloud/linux-64/rpodata.json>
Elapsed: -
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
ConnectTimeout(MaxRetryError("HTTPSConnectionPool(host='nanomirrors.tuna.tsinghua.edu.cn', port=443): Max retries exceded with url: /anaconda/cloud/linux-64/repodata.json (Caused by ConnectTimeoutError(<requests.packages.urllib3.connecton.VerifiedHTTPSConnection object at 0x7fb6d340dcc0>, 'Connection to nanomirrors.tuna.tsinghua.edu.cn timed out. (connct timeout=9.15)'))",),)

请检查

gedit ~/.condarc

将里面的东西全部删除保存，然后再执行安装命令即可（看别人的博客都是添加各种镜像去配置，我也试过了不行，删除却ok了）