在使用专业数据库软件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元/月，提供免费备案。另外提供免费虚拟主机，可供测试，让您快速上线网站。
选购地址：https://www.kufanyun.com/host/
图片来源于AI模型，如侵权请联系管理员。作者：路飞，如若转载，请注明出处：https://www.kufanyun.com/ask/1041.html
