深度学习GPU版本的pytorch安装

默认已经安装好了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