RuntimeError:CUDA error:out of memory // cmd》nvidia-smi 查看gpu-memory现在的状态 import os os.environ["CUDA_VISIBLE_DEVICES"] = '1'