Pytorch (1):GPU 查询

作者: 馨不累 | 来源:发表于2020-07-27 15:23 被阅读0次

torch.cuda.is_available()cuda是否可用；

torch.cuda.device_count()返回gpu数量；

torch.cuda.get_device_name(0)返回gpu名字，设备索引默认从0开始；

torch.cuda.current_device()返回当前设备索引；

image.png

网友评论

本文标题：Pytorch (1):GPU 查询

本文链接：https://www.haomeiwen.com/subject/urvzlktx.html

备案信息：桂公网安备 45052102000051号 · 桂ICP备13007215号-3

本站所收录作品、热点评论等信息部分来源互联网，目的只是为了系统归纳学习和传递资讯

所有作品版权归原创作者所有，与本站立场无关，如不慎侵犯了你的权益，请联系我们告知，我们将做删除处理！