admin 管理员组

文章数量: 887016

1、PyTorch的官方Windows支持

2018年4月25日，PyTorch 官方发布 0.4.0 版本，该版本的PyTorch 有多项重大更新，其中最重要的改进是支持 Windows系统。Windows用户能直接通过conda、pip和源码编译三种方式来安装Pytorch，不过需要强调Windows下的Pytorch仅支持Python3.5和Python3.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流程

本文标签： 版本 pytorch GPU

版权声明：本文标题：windows7下安装pytorch 0.4.0 GPU版本 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/jishu/1729680394h1338719.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。