在 Google Colab 上请求特定类型的 GPU

Question

我正在使用 Google Colab 运行一些深度学习代码。

前几天它可以正常运行，但最近，在不更改代码的情况下，我收到了错误：

RuntimeError: CUDA call failed (correlation_forward_cuda at correlation_cuda.cc:80)

在 Google Colab 上，我刚刚重新启动了运行时并再次尝试了代码。这次它成功了 - 所以我只是怀疑这取决于运气 - 取决于你是否获得了兼容的 GPU。

问题是，现在每次运行代码时，我都会收到错误消息，无论我重新启动运行时或关闭笔记本或将运行时类型更改为“无”然后返回到 GPU 有多少次 - 我现在总是收到错误消息。

有没有办法可以请求特定类型的 GPU？我还不确定是什么类型 - 但肯定有一个版本在某一时刻成功运行了我的代码。

我的代码需要一段时间才能达到故障点，因此一遍又一遍地运行代码并希望它能工作一次是非常令人沮丧的。

Answer 1

您无法选择特定的 GPU。连接到虚拟机后，您可以使用以下命令检查分配的 GPU：

!nvidia-smi --query-gpu=gpu_name,driver_version,memory.total --format=csv

如果给定的不是您想要的，您可以在“管理会话”菜单中终止计算机（如果按钮不起作用，请刷新页面）。如果你幸运的话，通过再次连接到另一台机器，你将获得你想要的 GPU，否则你可以重复这个过程直到那时。

我发现您发布的链接与 DAIN 应用程序相关。我不知道具体情况，也不知道您是否正在尝试执行它，但我在 Colab 上尝试了 DAIN 网络，并且仅适用于 P1000 和 P4。T4 和 K80 未能抛出您提到的相同错误。