pytorch官网:https://pytorch.org/
这里安装pytorch0.4.1版本(最新版本为1.3.0系列,但是在跑github上的一些项目时会不断地报“
UseWarinig:Legacy autograd function with non-static forward method is deprecated and will be removed in 1.3”的警告log,很影响观察,所以我们降级为了0.4.1版本;
1、pytorch0.4.1(cuda90):
(1)在https://download.pytorch.org/whl/cu90/torch_stable.html中找到对应版本,下载下来;
(2)到刚才下载的位置,使用pip install xxx.whl进行安装;
2、torchvision0.2.1:
注意:torchvision需要安装0.2.1系列的,否则会报错!!!安装方法:pip install torchvision==0.2.1