pytorch0.4.1安装

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

 

posted @ 2018-12-14 11:03  outthinker  阅读(3167)  评论(0编辑  收藏  举报