1. 内存不够 解决方法:尽量把显存的内存释放掉,具体可以通过 gc 进行内存回收,以及把显存释放。 del X_train, y_train, X_valid, y_valid gc.collect() torch.cuda.empty_cache() # torch.cuda.clear_memory_allocated() # entirely clear all allocated memory