一、进入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加速