!
也想出现在这里? 联系我们
广告位
当前位置:首页>教程分享>服务器教程>如何使用pytorch 查看 GPU 信息?

如何使用pytorch 查看 GPU 信息?

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

如何使用pytorch 查看 GPU 信息?

PyTorch是什么?

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

使用pytorch 查看 GPU 信息:

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

In [1]: import torchIn [2]: torch.cuda.current_device()Out[2]: 0In [3]: torch.cuda.device(0)Out[3]: In [4]: torch.cuda.device_count()Out[4]: 1In [5]: torch.cuda.get_device_name(0)Out[5]: \'GeForce GTX 950M\'In [6]: torch.cuda.is_available()Out[6]: True

给TA打赏
共{{data.count}}人
人已打赏
服务器教程

如何查看mysql版本的四种方法

2021-4-17 20:11:04

服务器教程

云服务器选择Linux系统好用还是选择Windows系统?

2021-4-17 20:11:06

声明 本站上的部份代码及教程来源于互联网,仅供网友学习交流,若您喜欢本文可附上原文链接随意转载。无意侵害您的权益,请发送邮件至 [email protected] 或点击右侧 私信:吉吉国王 反馈,我们将尽快处理。
0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索
OneEase