指定GPU運行python程序

1、命令行運行python程序時

一、首先查看哪些GPU空閒，nvidia-smi顯示當前GPU使用狀況。

nvidia-smi 

二、而後指定空閒的GPU運行python程序。

CUDA_VISIBLE_DEVICES=0,2,3 python test.py 

2、在python程序中指定GPU

import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" 

PS：週期性輸出GPU使用狀況

每 10s 顯示一次GPU使用狀況：

watch -n 10 nvidia-smi