直接去pytorch官网 https://pytorch.org/ ,根据自己的电脑配置选择对应的选项,选好之后,会自动生成适合你自己电脑的安装命令,如下所示:
然后在自己的环境下运行这个命令即可。
需要注意的几点:1、自己若要用虚拟环境,则需要先在anaconda里面创建一个虚拟环境,使用下面的命令:conda create --name pytorch python=3.6,这样就建成一个python版本为3.6的pytorch环境。
2、接着激活环境,使用 activate pytorch
3、在激活的环境中运行上面自动生成的命令即可。注意如果电脑上pip3报错,使用pip即可。
之后就完成了安装。打开python编译器,导入模块 import torch、import torchvision 不报错即可。