GPU常用命令

在终端执行程序时指定GPU

$ CUDA_VISIBLE_DEVICES=1   python  run_file.py

查看一台服务器上所有显卡上的使用者和GPU占用情况

首先安装gpustat
pip install gpustat

安装完成后使用下面命令即可

watch -n 2 --color gpustat --c

查看完成后 Crtl +c 即可退出

 
posted @ 2022-09-23 16:54  WangAlice  阅读(103)  评论(0)  编辑  收藏  举报