cupy.cuda.memory.OutOfMemoryError

Question

请教：
训练lda2vec的时候

gpu_id = int(os.getenv('CUDA_GPU', '0'))
cuda.get_device(gpu_id).use()
print("Using GPU " + str(gpu_id))

训练几十个epoch以后cupy.cuda.memory.OutOfMemoryError，然后又重头开始训练

data = prepare_topics(cuda.to_cpu(model.mixture.weights.W.data).copy(),
                          cuda.to_cpu(model.mixture.factors.W.data).copy(),
                          cuda.to_cpu(model.sampler.W.data).copy(),
                          words)

Answer 1

内存不够，你的内存是不是遇到了2GB上限？这可能是因为你的pycharm或者python解释器是32bit的，你可以换notebook和64bit python
如果还遇到问题，就是物理内存不足。