Pytorch windows10安装教程-阿里云开发者社区

开发者社区> 人工智能> 正文
登录阅读全文

Pytorch windows10安装教程

简介: 强烈建议安装anaconda之后再来安装这个pytorch,具体怎么安装百度搜索就知道了。 温馨提示,在安装anaconda的时候记得将“添加到环境变量”(安装的时候是英文的)这一选项选上。 下面假设你已经安装好anaconda了: 1.

强烈建议安装anaconda之后再来安装这个pytorch,具体怎么安装百度搜索就知道了。

温馨提示,在安装anaconda的时候记得将“添加到环境变量”(安装的时候是英文的)这一选项选上。

下面假设你已经安装好anaconda了:

  • 1.第一步下载pytorch的安装包:

链接: https://pan.baidu.com/s/1mh6U01i 密码: 8yty

  • 2.打开命令行进入上面所下载的安装包所在的目录,然后输入如下命令:
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2
  • 3.接下来就是查看是否安装成功
    进入python命令行交互界面输入如下
ipython
Python 3.6.2 |Anaconda custom (64-bit)| (default, Sep 19 2017, 08:03:39) [MSC v.1900 64 bit (AMD64)]
Type 'copyright', 'credits' or 'license' for more information
IPython 6.1.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import torch
In [2]:

若没报错则表示安装成功

  • 4.查看CUDA和cudnn是否能正常工作
In [6]: x = torch.Tensor([666])

In [7]: x_cuda = x.cuda()

In [8]: print(x_cuda)

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


In [9]: print(cudnn.is_acceptable(x_cuda))
True

这样表示cuda和cudnn可以正常使用。

如果cudnn报错,则使用如下代码将其关闭

cudnn.enabled = False


MARSGGBO原创





2017-10-2



版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

分享:
人工智能
使用钉钉扫一扫加入圈子
+ 订阅

了解行业+人工智能最先进的技术和实践,参与行业+人工智能实践项目

其他文章