代码中添加: import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0" //这里是自己的GPU编号,我是"0",原文是"8". 来源:CSDN作者:RainingZ链接:https://blog.csdn.net/qq_23096723/article/details/103243774 标签 显存