torch.cuda命令查询指的是什么
今天就跟大家聊聊有关torch.cuda命令查询指的是什么,可能很多人都不太了解,为了让大家更加了解,小编给大家总结了以下内容,希望大家根据这篇文章可以有所收获。
定制网站建设可以根据自己的需求进行定制,成都做网站、网站设计构思过程中功能建设理应排到主要部位公司成都做网站、网站设计的运用实际效果公司网站制作网站建立与制做的实际意义
1. 服务器GPU状态查询
1) lspci | grep -i nvidia 可以查询所有nvidia显卡
2) lspci -v -s [显卡编号] 可以查看显卡具体属性
3) nvidia-smi 可以查看显卡的显存利用率
2. torch.cuda主要函数
1) 查看是否有可用GPU: torch.cuda.is_available()
可用GPU数量:
gpus = torch.cuda.device_count() logger.info(f'use {gpus} gpus') logger.info(f"args: {args}")
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)
看完上述内容,你们对torch.cuda命令查询指的是什么有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注创新互联行业资讯频道,感谢大家的支持。
文章题目:torch.cuda命令查询指的是什么
本文链接:http://pwwzsj.com/article/ipsgsc.html