import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
import os os.environ["CUDA_VISIBLE_DEVICES"] = "2"
python中使用指定GPU
原文：https://www.cnblogs.com/sbj123456789/p/9876474.html
