本文主要致力于提供pytorch的解决方法。
我是在anaconda的环境下安装的。如果没有安装anaconda的可以先去官网安装个anaconda:anaconda官网。
1.正常安装pytorch
先去官网pytorch官网找到要下载的对应版本。
这里建议安装stable的版本。还要注意需不需要GPU加速,需要的话要看自己电脑支持的是哪个版本,可以去NVIDIA设置中查看,我这里选择的是10.2。
用anaconda安装的话建议使用conda进行安装(pip我试了下。。好像不行。。)
选择好了以后就复制下面的command。
打开Anaconda Prompt,将command粘贴,回车。
这个方法非常的慢,我试了好几次都因为太慢了中断了好几次。
然后我查到可以使用清华镜像源进行安装,速度会比刚才这个方法快上很多!
2.使用清华镜像源安装pytorch
第一步:打开Anaconda Prompt。
第二步:更改channels设置,依次执行以下命令:
第三步:我们还是到pytorch的官网,复制对应版本的command,然后删除后面的-c pytorch,因为-c pytorch默认从pytorch官网下,是非常慢的。删除-c pytorch之后,执行该命令,可以发现这个方法安装要快上很多。
下载完之后我们可以验证一下是否下载成功:
可以在当前环境下输入python命令,然后输入import torch看是否报错;我们用conda list命令也可以查看是否安装成功,可以发现里面已经有pytorch及torchvision等等。