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

Question

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

在前几天它可以正常运行，但最近，没有对代码进行更改，我得到了错误:

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

Apparently this is an error that can appear depending on the graphics card or system that you are using to run the code.

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

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

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

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

Answer 1

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

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

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

我看到您发布的链接与 DAIN 应用程序有关。我不知 Prop 体案例，或者您是否正在尝试执行它，但我在 Colab 上尝试了 DAIN 网络，并且只与 P1000 和 P4 一起使用。 T4 和 K80 未能抛出您所指的相同错误。