windows7下安装pytorch 0.4.0 GPU版本

1、PyTorch的官方Windows支持

2018年4月25日,PyTorch 官方发布 0.4.0 版本,该版本的PyTorch 有多项重大更新,其中最重要的改进是支持 Windows系统。Windows用户能直接通过condapip和源码编译三种方式来安装Pytorch,不过需要强调Windows下的Pytorch仅支持Python3.5Python3.6,不支持其他的Python3版本,也不支持Python2。我们在官网就能看到相应的安装方式:


但是,官网的方式是需要翻墙才能的,我用官网的方式安装时,在中途断掉了,所以后面采用了离线安装的方式。

2、Pytorch 0.4.0文件下载

首先说明:本文电脑现有的配置是windows7 64位,已经安装了Anaconda3,cuda 8.0 和cudnn(具体多少忘了)。所以针对现有的条件,我们在官网选择相应的配置(需翻墙)如下图所示:


然后复制上面画了红线的URL到浏览器,就会自动下载了。

3、Pytorch 0.4.0 安装

  • 下载完成之后,打开cmd;
  • cd到下载文件的文件夹中,本人下载到了:C:\Users\hp\Downloads;
  • 然后输入以下命令,总流程如下图所示:


4、Pytorch 0.4.0 测试

安装完成之后,我们需要测试Pytorch是否安装成功了。步骤如下:

  • 打开cmd,并输入python(这里相当于就是打开一个工作环境);


  • 导入pytorch,打印pytorch的版本信息,输入以下内容:
import torch
print(torch.__version__)

  • 新建一个tensor变量,并将它转换成GPU变量:
x = torch.Tensor([1.0])           # 声明一个变量
xx = x.cuda()                     # 将变量转换到GPU上
print(xx)

  • 检验cudnn是否安装成功:
from torch.backends import cudnn         # 导入模块
cudnn.is_acceptable(xx)

  • 总的测试流程和结果图如下:

说明:

  1. 关于Anaconda、cuda 8.0 、cudnn如何安装的可以参考本人以前的博客或他人的博客;
  2. 如有问题欢迎大家留言讨论。

参考:

[1] Windows下安装PyTorch0.4.0

[2] Win10 Python3.6下安装PyTorch

[3] Ubuntu16.04安装GPU版Pytorch流程

猜你喜欢

转载自blog.csdn.net/u011622208/article/details/80275344