torch.cuda命令查询

1. 服务器GPU状态查询

    1) lspci | grep -i nvidia 可以查询所有nvidia显卡

    2) lspci -v -s [显卡编号] 可以查看显卡具体属性

    3) nvidia-smi 可以查看显卡的显存利用率

2. torch.cuda主要函数

    1) 查看是否有可用GPU、可用GPU数量: torch.cuda.is_available(), torch.cuda.device_count()

    2) 查看当前使用的GPU序号:torch.cuda.current_device()

    3) 查看指定GPU的容量、名称:

        torch.cuda.get_device_capability(device),  torch.cuda.get_device_name(device)

    4) 清空程序占用的GPU资源: torch.cuda.empty_cache()

5) 为GPU设置随机种子:torch.cuda.manual_seed(seed), torch.cuda.manual_seed_all(seed)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容