在windows环境下安装pytorch

对于Windows的吐槽直接就忽略吧,切入主题。

本机环境

  1. 64-bit
  2. windows7
  3. no CPU

最好的安装方法

  1. install ANACONDA (附上ANACONDA windows Install link)

    • 由于pytorch 在Windows不支持Python 2.7版本,请毫不犹豫下载Anaconda 5.1 的Python 3.6 version。
    • 安装的时候除了全部都是default,最后一个界面是问你要不要了解一下ANACONDA,看心情选择了解与否。
    • 安装需要admin 权限(告知没有权限的亲)。
  2. 安装好后将\path\to\installed_location\Continuum\anaconda3,
    以及path\to\installed\location\Continuum\anaconda3\Scripts加入到PATH,以便未来方便使用

    • PATH位于:“Computer > System properties > Advanced system setting”
    • 本步骤需要admin权限
    • 开一个Command Prompt, run “conda -V”。如有version信息返回,则为正确。
  3. 去到pytorch界面,选择好和你自己环境相关的选项信息,然后run相应的command。

    • 无需纠结pip还是pip3。在anaconda3环境管理下,pip和pip3是一个东西。当然不能全信,可以run “pip -V”验证,如果version显示为anaconda3环境管理下的python3环境即为正确。
    • 有时候会被告知安装的pip版本过旧,请根据自动提示信息更新。
    • 有可能dependence缺少,请依据环境信息更新。
  4. 安装好后,在command prompt 内 run “py -3”以运行python3。接着run “import torch”。
    如无问题即完成安装。

    • 如果找不到Torch module,请尝试以下步骤:
      1. conda update conda
      2. conda install mkl=2018

遇到的坑

之前希望直接安装python3.5 然后再安装pytorch。成功安装pytorch后,运行时候被告知:

from torch._C import *
ImportError: DLL load failed: The specified module could not be found


查了类似错误发现是需要补救一些library给Windows。按照网上前辈的方法找好放好后还是不行。不知道是python3.5的坑还是我这台电脑大环境的坑,也未尝试升级python3.6后会不会有改变。但是还是conda安装简便靠谱,因为conda环境里面已经埋平了一些坑