其实主要还是之前配置Caffe配出心理阴影了。导致现在配个框架总是怕怕的。没想到现在的框架那么好配了。

  由于到目前Pytorch0.3.0在Windows下只支持CUDA9,还不支持CUDA8。而我这安装的就是CUDA8。所以安装的是0.2.1.

  其实感觉这配置,自己也是整的稀里糊涂的,不过总算是配置好了。

 



  谢谢各位博主的分享,以及谢谢网友  @data_coder @杨朝晖 @翔翔 @OLDPAN的帮助。

  整体总结:

1)其实配置这玩意,感觉还是有点要看人品的。不过前期要搞清楚,各种环境(软件型号、版本)之类的对应的问题。这样才能一遍过。

2)我主要安装的问题就是VS2015太难安装了。还有两个不同版本的Anaconda不知道咋处理。

3)

4)

5)

  其实之前已经配置VS2013、CUDA8和Anaconda2,但是发现环境不匹配,需要(Visual Studio 2015 with Update 2 及以上 和 Python 3.6.1以上)所以就需要重新安装一些软件。

  • 1、安装VS2015企业版,安装CUDA

  一开始想直接就安装VS2017了,但是网上找到的VS2017迅雷下载不下来。所以就下载了一个VS2015,发现真是坑呀。安装巨慢。后来发现这VS2015不能选择全部安装,按照默认安装就好了。但是,由于取消过安装的缘故,之前安装的VS2013,被搞坏了。顺带又卸载了之前安装的VS2013(其实这时,内心隐隐有点感觉不妙)。就VS2015,就安装了一整天呀。

  由于重新了VS,所以需要重新安装CUDA。

  • 2、下载Anaconda3,安装

  1. Windows (7/8/8.1/10)64位 + Anaconda-4.3.1 (64-bit) Python 3.6.0 
  2. Windows (7/8/8.1/10)64位 + Anaconda-4.4.0 (64-bit) Python 3.6.1 
  3.

  由于作者要求要3.6.1以上,去官网上下载的Anaconda3-5.0.1-Windows-x86_64进行安装。

  其实真没想到这感觉又会出问题。由于之前安装过Anaconda2,由于我基本放弃Caffe了。所以感觉可以先把之前安装的Anaconda2卸载掉。然后再安装Anaconda3.

  而且安装过程中,注意把下面这两个选项全都勾上最好。

  

cuda版本和pytorch对应关系_CUDA

  • 3、下载对应的Pytorch

  在百度云下载自己对应的Pytorch

  

cuda版本和pytorch对应关系_Python_02

  注:文件名说明:

  一般为PACKAGENAME-VERSION-PYTHON_VERSIONcu80.tar.bz

  PACKAGENAME 分为 pytorch 和 pytorch_legacy, 分别为NT内核版本10和6的两类系统进行编译;VERSION 代表 pytorch 的版本;而PYTHON则代表python程序的版本,主要分为3.5和3.6;请自行选择合适的版本进行安装。

  由于我的内核版本为6,而且python为3.6.所以对应的就是

cuda版本和pytorch对应关系_Python_03

  • 4、安装PyTorch

  下载完毕之后,直接cmd进入到pytorch对应的目录,执行以下命令:

install numpy mkl cffi

install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 #离线安装

  

cuda版本和pytorch对应关系_Windows_04

  如果conda不行,也可以pip安装

.tar.bz2

  • 5、测试PyTorch

  

cuda版本和pytorch对应关系_Python_05

  安装完成。

  • 6、安装torch-vision

  根据官网

 

cuda版本和pytorch对应关系_CUDA_06

  发现conda不行,还是要pip

  

cuda版本和pytorch对应关系_Windows_07