Windows10下安装Pytorch0.3
环境准备
Anaconda3 (with Python 3.5/3.6)
Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
CUDA 8 / CUDA 9(包中已内置,因此老版本也能用)
cuDNN v5以上(包中已经自带了, cuDNN v5用户亦可放心使用)
如果不想放弃Anaconda2也可以创建虚拟环境来使用Anaconda3。
Anaconda2与Anaconda3共存方法,详细参考:http://blog.****.net/infin1te/article/details/50445217
安装好Anaconda3后activate Anaconda3,再conda create –name pytorch,为pytorch创建一个虚拟环境。
Pytorch安装
-
**pytorch
activate Anaconda3
activate pytorch
由于是在Anaconda2下的Anaconda3中新建的虚拟环境,所以每次**pytorch时,需要先**Anaconda3。 -
安装依赖项
conda install numpy mkl setuptools cmake cffi
conda install -c peterjc123 vc
conda install -c peterjc123 cuda80 -
安装pytorch
conda install -c peterjc123 pytorch
由于网速太慢,在conda官网下载了需要的版本的pytorch,在本地安装。
conda install –offline \mypath\pytorch????.tar.bz2
安装的时候,报错:CondaError: EOFError(‘Compressed file ended before the end-of-stream marker was reached’,)
解决方法http://blog.****.net/okcd00/article/details/78022723
将Anaconda3\pkgs下的pytorch安装包删掉,运行成功。
4. 测试
import torch
没报错说明安装成功!