【使用指定GPU&实时监控GPU】

1、pytorch如何指定GPU

pytorch默认使用第0块GPU训练模型，如果第0块用满了，我们可以指定使用别的GPU进行训练

方法：
用这两行代码
需要放在整个.py文件的最开头，在调用torch之前就要用，不然无效

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

2、gpustat监控GPU状态

gpustat -i

Ctrl+c 退出监控

先使用上面的命令试一下，如果服务器维护人员没有安装，则直接使用下面命令即可安装。

pip install gpustat