win10安装pytorch

win10安装pytorch

首先确定自己电脑适用的版本

笔者电脑环境为：

• python3.7
• cuda9.2（可通过NVIDIA控制面板查询得到）

方法一：利用pip安装

在pytorch官网中，根据自己的电脑配置选择适合的命令，打开cmd输入命令进行安装。

方法二：使用conda安装

方法同上，不过选择Conda而不是Pip

由于在国内，上述两种方法同时无法成功

方法三：使用conda安装，但是通过清华镜像下载

我们需要在cmd中输入如下命令，将清华镜像添加到Anaconda的channel中

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

设置好之后，同样通过pytorch官网提供的命令进行安装：

在这里有个坑

官网提供的命令为：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

其中，-c pytorch参数指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库。
因此，只需要将-c pytorch语句去掉，就可以使用清华镜像源快速安装pytorch了。

但是，笔者使用这种方法同样出错。

方法四：手动下载whl文件，进行安装

在点这里中下载对应自己电脑配置的torch以及torchvision的whl文件

笔者下载的文件如下：

最后使用pip命令进行安装：

pip install torch-1.4.0+cu92-cp37-cp37m-win_amd64.whl
pip install torchvision-0.4.2+cu92-cp37-cp37m-win_amd64.whl

终于成功！