Windows支持Pytorch?如何在windows10中使用Pytorch?

earnpls • 19471 次浏览 • 0 个回复 • 2017年07月06日

Pytorch官网没有提供windows安装方法,Pytorch不支持windows,不过爱折腾的兄弟们怎么可能不把这个弄出来呢?你需要支持的条件:

  1. Anaconda3 (with Python 3.6)
  2. Windows 10 64位系统(其他未测试)
  3. CUDA 8.0.61
  4. cuDNN 5.1.10及以上
  5. Compute Capability 6.1及以上的显卡

注意

目前官方已经支持Windos版本的Pytorch 0.4.0版本,不再推荐使用本文章这样的安装方式,Pytorch 0.4.0需要在Python 3中运行,安装方法参考: Windos使用CONDA/PIP安装官方pythorch 0.4.0/torchvision

在win10 python3.6开启Pytorch支持:

安装方法一、

conda install -c peterjc123 pytorch=0.1.12

不过安装速度比较慢,如果不嫌慢的话就用方法二

安装方法二、

首先安装Anaconda3,然后下载Pytorchconda安装包,地址:pytorch conda安装包—pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 下载完毕之后,直接cmd进入到pytorch-0.1.12-py36_0.1.12cu80.tar.bz2对应的目录,执行以下命令:

conda install numpy mkl cffi
conda install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 #离线安装

然后python命令下输入:

import torch

如果没有报错,代表安装成功了,如下图所示:

Windos安装Pytorch

提示:经网友提醒,若import torch时发生如下错误:

Traceback (most recent call last):
  File "test.py", line 2, in <module>
    import torch
  File "C:\Anaconda3\lib\site-packages\torch\__init__.py", line 41, in <module>
    from torch._C import *
ImportError: DLL load failed: The specified module could not be found.

请将Anaconda的Python版本从3.6.0升级至3.6.1。

使用如下代码测试CUDA与cuDNN是否工作正常的代码:

# CUDA 测试
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

如果CUDA不正常工作,代表不能使用.cuda()将模型和数据通过GPU进行加速。而如果cuDNN不能正常工作,那就使用如下代码关掉它:

cudnn.enabled = False

原创文章,转载请注明 :Windows支持Pytorch?如何在windows10中使用Pytorch? - pytorch中文网
原文出处: https://www.ptorch.com/news/20.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! Pytorch v0.1.12版本发布,添加CUDA的Sparse支持