深度学习 VM:如何释放 RAM?

Deep Learning VM: How to free RAM?

我是 GCP 和深度学习 VM 的新手。我得到它来训练一些深度学习模型。在 google 云 jupyter 笔记本中进行训练时,它崩溃了,因为它无法将输入张量从 GPU 复制到 CPU:具体来说:

InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:GPU:0 to /job:localhost/replica:0/task:0/device:CPU:0 in order to run TensorDataset: Dst tensor is not initialized. [Op:TensorDataset]

查了一下,是GPU内存不足的时候出现的。我检查了我的内存,在我初始化 VM 后 运行 大约一个小时后,我的 RAM 已满 95%。我不知道这是怎么发生的。我怎样才能释放这个内存?

发现即使在 python 脚本终止后,GPU 中的内存仍然存在。 运行 nvidia-smi 查看是否有 python 进程占用 GPU 内存,如果有 运行 pkill -9 python 杀死所有相关内存和进程 python.