Windows下Pytorch安装过程

环境版本

Windows： 10
Anconda： 4.8.3
CUDA： 10.0
CUDNN： 7.6.5
Python： 3.6
Pytorch: 1.2.0

参考资料

windows10 Anaconda安装pytorch(GPU)踩坑！

安装流程

1. 查看CUDA型号
2. 进入清华镜像网站
   清华镜像网站
   选择win-64
   下载pytorch-1.2.0-py3.6_cuda100_cudnn7_1.tar.bz2
   下载torchvision-0.4.0-py36_cu100.tar.bz2
3. 在anaconda中新建一个环境，其中Python版本为3.6

conda create --name PYT python=3.6

4. 进入环境

activate PYT

5. 配置清华镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

6. 安装numpy等包
   mkl比其余两个包要大，所以可能会下载失败，只要重复运行这行代码就可以了，即多次下载，直至成功

conda install numpy mkl cffi

7. 切换到前面下载文件所在的文件夹路径

cd C:\Users\DELL\Downloads\Compressed

8. 安装pytorch和torchvision

conda install --offline pytorch-1.2.0-py3.6_cuda100_cudnn7_1.tar.bz2
conda install --offline torchvision-0.4.0-py36_cu100.tar.bz2

9.进入python测试是否安装成功

import torch
import torchvision
print(torch.__version__) print(torchvision.__version__)
torch.cuda.is_available()

也许会在import torchvision这一步报错，报错内容为

ModuleNotFoundError: No module named ‘PIL’

此时只需要pip install -i https://pypi.douban.com/simple pillow
测试结果

>>> import torch
>>> import torchvision
>>> print(torch.__version__)
1.2.0
>>> print(torchvision.__version__)
0.4.0
>>> torch.cuda.is_available()
True