2021年10月12日
pytorch中查看gpu信息
摘要： torch.cuda.is_available()cuda是否可用； torch.cuda.device_count()返回gpu数量； torch.cuda.get_device_name(0)返回gpu名字，设备索引默认从0开始； torch.cuda.current_device()返回当前设 阅读全文
posted @ 2021-10-12 15:17 呦呦南山 阅读(557) 评论(0) 推荐(0)
实时查看GPU显存使用情况
摘要： 该指令可实时查看gpu使用情况 watch -n1 gpustat 阅读全文
posted @ 2021-10-12 11:54 呦呦南山 阅读(740) 评论(0) 推荐(0)