python指定GPU运行程序

import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “2”

你可能感兴趣的:(python指定GPU运行程序)