专注、交流、分享
云计算领域最新资讯

如何使用pytorch 查看 GPU 信息?

在使用专业数据库软件PyTorch时都会需要查看GPU信息,很多小伙伴不知道PyTorch如何查看GPU,下面让我们来看看如何使用pytorch 查看 GPU 信息?

PyTorch是什么?

PyTorch是一个开源Python机器学习库,基于Torch,用于自然语言处理等应用程序,前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。是一个以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络。

使用pytorch 查看 GPU 信息:

输入下面代码就可以查看GPU信息

In [1]: import torch

In [2]: torch.cuda.current_device()
Out[2]: 0

In [3]: torch.cuda.device(0)
Out[3]: <torch.cuda.device at 0x7efce0b03be0>

In [4]: torch.cuda.device_count()
Out[4]: 1

In [5]: torch.cuda.get_device_name(0)
Out[5]: 'GeForce GTX 950M'

In [6]: torch.cuda.is_available()
Out[6]: True

以上是关于“如何使用pytorch 查看 GPU 信息。”的介绍,如需购买测试PHP主机,推荐酷番云共享虚拟主机、独享IP虚拟主机齐备,各类配置均有,满足不同网站建设需求;价格实惠;最低9元/月,提供免费备案。另外提供免费虚拟主机,可供测试,让您快速上线网站。

选购地址:kufanyun.com/host/

 

 

赞(1)
未经允许不得转载:酷番云知识库 » 如何使用pytorch 查看 GPU 信息?
  • 日销500+
    基础型虚拟主机

    共享CPU

    共享内存

    2G空间

    共享带宽

    共享IP

    分布式存储

    适合个人博客、个人站长类网站

    ¥9 原价¥28

  • 日销500+
    超值型云主机CVM

    2核CPU

    2G内存

    180G硬盘

    5M带宽

    独立IP

    分布式存储

    适合企业官网、行业门户类网站

    ¥99 原价¥210

  • 日销500+
    通用型云服务器ECS

    4核CPU

    4G内存

    50G硬盘

    20M带宽

    独立IP

    分布式存储

    适合电商、论坛类网站

    ¥254 原价¥322

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址

中国领先的企业级云服务提供商

域名注册云服务器