服务器GPU查看和设置

查看GPU占用情况

nvidia-smi
-周期性的输出显卡的使用情况:
watch -n 10 nvidia-smi

命令行参数-n后边跟的是执行命令的周期,以s为单位。

设置程序运行GPU

1.代码中加入

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

注意: 服务器看到的GPU序号和手动在Linux上查看的序号是相反的,比如
在这里插入图片描述

如果指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”,则在watch nvidia-smi指令下,2号GPU正在满负荷的运行!

2.命令行CUDA_VISIBLE_DIVICES=1 python xxx.py


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部