温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

torch.cuda命令查询指的是什么

发布时间:2021-11-11 17:56:10 来源:亿速云 阅读:158 作者:柒染 栏目:大数据

今天就跟大家聊聊有关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命令查询指的是什么有进一步的了解吗?如果还想了解更多知识或者相关内容,请关注亿速云行业资讯频道,感谢大家的支持。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI