解决报错:RuntimeError:CUDA out of memory

爆内存,屏蔽显卡,换用cpu训练

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

你可能感兴趣的:(python)