windows下安装pytorch1.7.1
背景
由于课题需要pytorch,所以转向pytorch。而网上的都下载特别慢,尤其是官网上面的,如果是用conda下载的话,清华源的镜像会特别不好使,但是如果用官网的话,下载会直接报错:
而官网上的要求安装的包如下:
但是咱们最后需要的包也就是torch、torchvision。所以直接安装这两个就可以了。
操作:
点击上面连接,找到这个:
注意:cu110指的是cuda版本是11.0版本的,cp37指的是python=3.7版本,win_amd64指的是windows版本64位。下载好了之后,在目录处输入cmd,pip install 下载的文件名。
安装成功后:
强烈建议提前使用pip安装numpy,安装太慢了。
同理再安装torchvision:
最后测试: