我可以在colab中清理GPU显存吗

Question

我正在尝试使用 aitextgen 在数据集上微调 774M gpt 2。不幸的是，无论我做什么，训练都会失败，因为只有 80 MB 的 vram 可用。如何在不重新启动运行时的情况下清除 vram，并可能防止 vram 已满？

Answer 1

另一种解决方案是使用这些代码片段。

1.

!pip install numba

2. 然后：

from numba import cuda
# all of your code and execution
cuda.select_device(0)
cuda.close()

你的问题在Tensorflow官方github上有讨论。 https://github.com/tensorflow/tensorflow/issues/36465

更新：@alchemy 报告说，在开启方面这是无法恢复的。
您可以尝试下面的代码。

device = cuda.get_current_device() 
device.reset()

Answer 2

1. 在笔记本块内运行命令 !nvidia-smi。
2. 查找您不需要删除以清理 vram 的 GPU 的进程 ID。然后运行命令 !kill process_id

它应该对你有帮助。

Answer 3

PyTorch 自动管理 CUDA 内存，因此您通常不需要手动关闭设备。如果您想释放 GPU 内存，可以尝试以下操作：

import torch

# Deletes all unused tensors
torch.cuda.empty_cache()