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:

在这里插入图片描述

最后测试:

在这里插入图片描述

原文地址:https://www.cnblogs.com/chenyameng/p/14273935.html