我想知道是否pytorch正在使用我的GPU.可以nvidia-smi在进程中检测GPU是否有任何活动,但我想要用python脚本编写的东西.
pytorch
nvidia-smi
python
有办法吗？
python memory-management gpu nvidia pytorch
gpu ×1
memory-management ×1
nvidia ×1
python ×1
pytorch ×1