【Pytorch】Windows10下配置Pytorch环境

2017-09-18 留言 环境配置

0x00 前言

前言什么的也懒得说了……
总之：
听说你Pytorch很牛，
不乐意让我Windows用，
而我又听说pytorch用来训练模型超好用，
不仅没头脑而且不高兴！我要在我的windows上配一个！

0x01 准备工作

0x02 配置开始

首先，我们需要一个比较配pytorch的环境，此处使用python3.6.1，因为我把离线安装包下载到了D盘的Downloads目录下，所以我们就在D:\Downloads\这里操作吧，根据你们想放在哪里自行安排。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
$ conda create -n torch python=3.6.1
Fetching package metadata ...........
Solving package specifications: .

Package plan for installation in environment C:\ProgramData\Anaconda3\envs\torch:

The following NEW packages will be INSTALLED:

    certifi:        2016.2.28-py36_0
    pip:            9.0.1-py36_1
    python:         3.6.1-2
    setuptools:     36.4.0-py36_0
    vs2015_runtime: 14.0.25420-0
    wheel:          0.29.0-py36_0
    wincertstore:   0.2-py36_0

Proceed ([y]/n)? y

vs2015_runtime 100% |###############################| Time: 0:01:28  23.07 kB/s
python-3.6.1-2 100% |###############################| Time: 0:28:32  19.29 kB/s
certifi-2016.2 100% |###############################| Time: 0:00:10  21.24 kB/s
wincertstore-0 100% |###############################| Time: 0:00:00  45.54 kB/s
setuptools-36. 100% |###############################| Time: 0:00:21  28.14 kB/s
#
# To activate this environment, use:
# > activate torch
#
# To deactivate an active environment, use:
# > deactivate
#
# * for power-users using bash, you must source
#

然后我们进入这个只有默认包的conda环境中，至少把numpy啥的pytorch必备组件给他装上：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
(torch) D:\Downloads>conda install numpy mkl cffi
Fetching package metadata ...........
Solving package specifications: .

Package plan for installation in environment C:\ProgramData\Anaconda3\envs\torch:

The following NEW packages will be INSTALLED:

    cffi:      1.10.0-py36_0
    mkl:       2017.0.3-0
    numpy:     1.13.1-py36_0
    pycparser: 2.18-py36_0

Proceed ([y]/n)?

pycparser-2.18 100% |###############################| Time: 0:00:06  27.88 kB/s

然后就是激动人心的装这个离线包了！

1
2
3
4
5
(torch) D:\Downloads>conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2

CondaError: EOFError('Compressed file ended before the end-of-stream marker was reached',)
CondaError: EOFError('Compressed file ended before the end-of-stream marker was reached',)
CondaError: EOFError('Compressed file ended before the end-of-stream marker was reached',)

我很想大声的喊一句“不可能只有我一个人，不会只有我遇到这种情况！”
在check了一下下载的压缩包并没有损坏之后，查到一个解决方案：（Thx to @李新鹏）:

解决‘CondaError: EOFError(‘Compressed file ended before the end-of-stream marker was reached’,)’问题，在anaconda/pkgs文件夹删除缓存的pytorch安装包再安装比较好，不会删去其余完好的安装包

然后在anaconda目录（比如我的在C:\ProgramData\Anaconda3\pkgs\）下，找到先前缓存的pytorch包（比如我的是C:\ProgramData\Anaconda3\pkgs\pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2），直接把这个文件删掉，然后回到CMD中，再次输入

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
(torch) D:\Downloads>conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2

(torch) D:\Downloads>python
Python 3.6.1 |Continuum Analytics, Inc.| (default, May 11 2017, 13:25:24) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import pytorch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch'
>>> import torch
>>> x = torch.Tensor([1.0])
>>> xx = x.cuda()
>>> print(xx)

 1
[torch.cuda.FloatTensor of size 1 (GPU 0)]

可喜可贺，可喜可贺~

啦啦啦

哦对了最后还可以放进jupyter notebook的kernel里用

1
2
3
sudo pip install -U ipykernel
# source activate torch
python -m ipykernel install --user
#PyTorch