天天看点

windows下pytorch安装过程

记录一下安装pytorch的流程。

必须得说一下,之前安装了很多python版本,安装pytorch忘了查python版本了,最后调了一堆问题,最后报错:

%1 不是有效的 Win32 应用程序,整个人要炸裂了,又得重新下载whl安装。

说一下,能用anaconda安装的最好用这个,能避免很多麻烦,这是想试试不用conda,用pip安装的流程。

1. 安装numpy+mkl

           确认Python版本3.6 64位

           卸载numpy   pip uninstall numpy

下载numpy+mkl的64位版本的whl  

网址:https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy

如我下的是numpy-1.14.6+mkl-cp36-cp36m-win_amd64.whl

一定要先安装numpy+mkl再安装torchvision,torchvision的下载依赖nump+mkl

要确认你的pip安装第三方库的路径是所用的python版本下。查询方式,pip show pip

见loaction,修改方式在系统环境变量中修改,把 python路径/script 置顶,  同样,默认的terminal的python也可在这里修改,不过倒是无所谓

          安装numpy+mkl的64位版本

          pip install D:/anaconda/download/numpy-1.14.6+mkl-cp36-cp36m-win_amd64.whl

2. 安装torch

下载torch的whl文件安装,

           如果用gpu,考虑下载gpu版本的,官网直接打开那个http即可,要先确认显卡的CUDA,查看方式,见截图

windows下pytorch安装过程

                  比如这里要用8.0,官网:https://pytorch.org/进入

windows下pytorch安装过程

把地址复制转到http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl就能下载,CUDA的比较大

同样保证64位

安装同上,pip install D:/anaconda/download/torch-0.4.1-cp36-cp36m-win_amd64.whl

3. 安装torchvision

         pip3 install torchvision

         这里pip运行会报timeout的错,不知道是网络问题还是pip版本问题,用pip3就没问题了

4. 测试

windows下pytorch安装过程

成功!

whl文件:

windows下pytorch安装过程

继续阅读