程序中指定GPU代码

import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “0, 1”