如何使用pytorch 查看 GPU 信息?

在使用专业数据库软件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]:<torch.cuda.deviceat0x7efce0b03be0>In[4]:torch.cuda.device_count()Out[4]:1In[5]:torch.cuda.get_device_name(0)Out[5]:\’GeForceGTX950M\’In[6]:torch.cuda.is_available()Out[6]:True

更多关于云服务器域名注册虚拟主机的问题,请访问西部数码官网:www.west.cn

赞(0)
声明:本网站发布的内容(图片、视频和文字)以原创、转载和分享网络内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-62778877-8306;邮箱:fanjiao@west.cn。本站原创内容未经允许不得转载,或转载时需注明出处:西部数码知识库 » 如何使用pytorch 查看 GPU 信息?

登录

找回密码

注册