设置可见GPU方式

只可见第一块GPU
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
原文地址:https://www.cnblogs.com/llfctt/p/10563595.html