WIN10中Anaconda安装gpu版本的pytorch

WIN10环境下安装pytorch：

尝试安装gpu版本的pytorch，尝试无数次以后，终于成功。

安装步骤：

1.首先查找自己gpu版本信息，在cmd中输入：

nvidia-smi

得到版本信息
我这里的CUDA支持11.1，在去pytorch官网看到可以安装10.1版本。
2.conda prompt中创建pytorch环境。

conda create --name torchenv python=3.6 

3.激活创建的环境

activate torchenv

4.改掉conda安装的镜像源，将以下复制进.condarc文件中。

  channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true
ssl_verify: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
        }

5.在pytorch官网找到安装语句，橙色的是根据自己CUDA版本选择的。

去掉 -c pytorch，不然无法从指定的清华源镜像下载。

遇到的问题：

1、一直出现
CondaHTTPError: HTTP 000 CONNECTION FAILED for url https:mirrors.tuna.tsinghua.edu.cnanacondacloudpytorchwin-64pytorch-1.6.0-py3.6_cuda101_cudnn7_0.tar.bz2多次更改镜像源，最终将源文件的https改成上面的http。
2、可能会出现下载超时继而失败，可以用下面的代码更改超时响应时间。

remote_read_timeout_secs: 600.0

最终结果：

torch.cuda.current_device()
0
torch.cuda.is_available()
True
torch.cuda.device_count()
1
torch.cuda.current_device()
0
torch.cuda.get_device_name(0)
'GeForce GTX 1660 Ti'