关于Pytorch的安装

一、进入pytorch官网，选择适合你的版本，如上图所示，复制给出的链接进行下载。
二、若是下载的太慢，那就用镜像下载。
比较常用的国内镜像包括：
（1）阿里云 http://mirrors.aliyun.com/pypi/simple/
（2）豆瓣http://pypi.douban.com/simple/
（3）清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
（4）中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
（5）华中科技大学http://pypi.hustunique.com/

你要尝试多个镜像进行安装，有时你所使用的镜像里，不一定有你要下载的库；亲测，阿里云能下载cuda=10.2，cudnn=7.6.5版本的pytorch。
镜像下载的方式很简单，就是多加上 -i http://mirrors.aliyun.com/pypi/simple/
例如：

pip install tensorflow-gpu -i http://mirrors.aliyun.com/pypi/simple/

pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html -i http://mirrors.aliyun.com/pypi/simple/

三、你可以先使用镜像下载，但有时在镜像里只能下载部分库。
实测，我用阿里云镜像下载pytorch，下载时会附带一些相关的库，然而有一个库，阿里云镜像是没有的，这就是再换镜像或者不用镜像了。
四、下载成果后的测试

import torch
torch.__version__  #返回pytorch的版本
torch.cuda.is_available()  # 返回True，则证明可以此pytorch可以使用cuda加速