默认已经安装好了cuda和anaconda(已经把源切换到国内)。两行指令轻松安装pytorch。
第一步:创建一个python虚拟环境,然后激活。
第二步:输入如下指令安装torch
pip install torch==1.2.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
其中torch==1.2.0是对应的torch版本
cu92是对应你电脑安装的cuda的版本
https://download.pytorch.org/whl/torch_stable.html是下载路径
第三步安装对应的torchvision
pip install torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html