Win7x64+cuda9+PyTorch0.3.0
其实本来感觉安装好了0.2就不想去折腾了。感觉那些bug对于我这样的屌丝机,无所谓。但是没想到在0.3版本下的源码在0.2运行会出错。所以没办法逼着升级。好在这安装非常简单。主要就是torch新版本会自动覆盖老版本。torchvision不需要升级。
主要 参考了:
谢谢蒲嘉宸的贡献,真是功德无量呀,以及谢谢网友 @nudt_qxx 的帮助。
整体总结:
1)
2)
3)
4)
5)
- 1、安装VS2015企业版,安装CUDA(后来发现其实VS2013就可以了,那就没有必要安装VS2015了,安装VS2015太麻烦了)
一开始想直接就安装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. Windows (7/8/8.1/10)64位 + Anaconda-5.0.1 (64-bit) Python 3.6.3
由于作者要求要3.6.1以上,去官网上下载的Anaconda3-5.0.1-Windows-x86_64进行安装。
其实真没想到这感觉又会出问题。由于之前安装过Anaconda2,由于我基本放弃Caffe了。所以感觉可以先把之前安装的Anaconda2卸载掉。然后再安装Anaconda3.
而且安装过程中,注意把下面这两个选项全都勾上最好。
- 3、下载对应的Pytorch
在百度云下载自己对应的Pytorch
注:文件名说明:
一般为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.所以对应的就是,现在网上下载的都是cuda9,cuda9向下兼容cuda8。所以只有这个包了。
- 4、安装PyTorch
下载完毕之后,直接cmd进入到pytorch对应的目录,执行以下命令:
conda install numpy mkl cffi
conda install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 #离线安装
如果conda不行,也可以pip安装
pip install pytorch-***.tar.bz2
- 5、测试PyTorch
安装完成。
- 6、安装torch-vision
根据官网:其实如果只是升级torch版本,那就不需要重新安装torchvison了。这个不需要升级。
发现conda不行,还是要pip