一、命令行运行python程序时
1、首先查看哪些GPU空闲，nvidia-smi显示当前GPU使用情况。
|
1
|
nvidia-smi
|
2、然后指定空闲的GPU运行python程序。
|
1
|
CUDA_VISIBLE_DEVICES=0,2,3 python test.py
|
二、在python程序中指定GPU
|
1
2
|
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3"
|
PS：周期性输出GPU使用情况
每 10s 显示一次GPU使用情况：
|
1
|
watch -n 10 nvidia-smi
|
想了解更多关于指定GPU运行python程序的内容，请扫微信
或微信搜索jiemingpan
版权说明：本文版权由作者自行负责，如有侵权请联系本站删除。