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.csdn.net/infin1te/article/details/50445217

安装好Anaconda3后activate Anaconda3,再conda create –name pytorch,为pytorch创建一个虚拟环境。

Pytorch安装

  1. 激活pytorch

    activate Anaconda3
    activate pytorch
    由于是在Anaconda2下的Anaconda3中新建的虚拟环境,所以每次激活pytorch时,需要先激活Anaconda3。

  2. 安装依赖项
    这里写图片描述

    conda install numpy mkl setuptools cmake cffi
    conda install -c peterjc123 vc
    conda install -c peterjc123 cuda80

  3. 安装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.csdn.net/okcd00/article/details/78022723
将Anaconda3\pkgs下的pytorch安装包删掉,运行成功。
4. 测试

import torch

没报错说明安装成功!

猜你喜欢

转载自blog.csdn.net/qq_23142123/article/details/78895854