扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
在使用专业数据库软件PyTorch时都会需要查看GPU信息，很多小伙伴不知道PyTorch如何查看GPU，下面让我们来看看如何使用pytorch查看 GPU 信息？
PyTorch是什么？
PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序，前身是Torch，其底层和Torch框架一样，但是使用Python重新写了很多内容，不仅更加灵活，支持动态图，而且提供了Python接口。是一个以Python优先的深度学习框架，不仅能够实现强大的GPU加速，同时还支持动态神经网络。
使用pytorch 查看 GPU 信息：
输入下面代码就可以查看GPU信息
In[1]:importtorchIn[2]:torch.cuda.current_device()Out[2]:0In[3]:torch.cuda.device(0)Out[3]:
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网