一、命令行運行python程序時
1、首先查看哪些GPU空閑,nvidia-smi顯示當前GPU使用情況。
nvidia-smi
2、然后指定空閑的GPU運行python程序。
CUDA_VISIBLE_DEVICES=0,2,3 python test.py
二、在python程序中指定GPU
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3"
PS:周期性輸出GPU使用情況
每 10s 顯示一次GPU使用情況:
watch -n 10 nvidia-smi