当前位置:首页 > 互联网 > 正文

如何使用pytorch 查看 GPU 信息?

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

PyTorch是什么?

如何使用pytorch 查看 GPU 信息? 第1张

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

使用pytorch 查看 GPU 信息:

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

如何使用pytorch 查看 GPU 信息? 第2张

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/

如何使用pytorch 查看 GPU 信息? 第3张

0