windows下安装pytorch1.7.1

windows下安装pytorch1.7.1

---

背景

由于课题需要pytorch，所以转向pytorch。而网上的都下载特别慢，尤其是官网上面的，如果是用conda下载的话，清华源的镜像会特别不好使，但是如果用官网的话，下载会直接报错：


而官网上的要求安装的包如下：


但是咱们最后需要的包也就是torch、torchvision。所以直接安装这两个就可以了。

---

操作：

pytorch下载包

点击上面连接，找到这个：


注意：cu110指的是cuda版本是11.0版本的，cp37指的是python=3.7版本，win_amd64指的是windows版本64位。下载好了之后，在目录处输入cmd，pip install 下载的文件名。

安装成功后：


强烈建议提前使用pip安装numpy，安装太慢了。

同理再安装torchvision：


最后测试：