Pytorch中查看GPU信息

本文摘自：知乎 用Pytorch中查看GPU信息

1. 返回当前设备索引

torch.cuda.current_device()

2. 返回GPU的数量

torch.cuda.device_count()

3. 返回gpu名字，设备索引默认从0开始

torch.cuda.get_device_name(0)

4. cuda是否可用

torch.cuda.is_available()

posted @   Picassooo  阅读(359)  评论(0)  编辑  收藏  举报
相关博文：
· pytorch、torchvision、detectron2等包的wheel下载地址
· pytorch GPU torch.nn.DataParallel (DP) 多卡并行
· PyTorch获取GPU信息（设备id、名称、显存）
· Python 利用 Pytorch 获取当前GPU信息
· torch查看gpu信息
阅读排行：
· Manus重磅发布：全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
历史上的今天：
2021-06-16 怎么写论文摘要
点击右上角即可分享
微信分享提示