指定运行代码的GPU

import os
os.environ[‘CUDA_VISIBLE_DEVICES’]='2’表示仅用2号GPU，‘3，4’表示用3和4号GPU。