Google Colab + Pytorch:运行时错误:没有可用的 CUDA GPU

时间:2021-05-09 02:07:04

标签: python-3.x pytorch google-colaboratory

错误截图:

screenshot of error

您好,我正在尝试运行 this Pytorch application,这是一个用于对狗和猫图片进行分类的 CNN。

我将 Google Colab 用于 GPU,但由于某种原因,我得到 RuntimeError: No CUDA GPUs are available。这很奇怪,因为我专门在 Colab 设置中启用了 GPU,然后测试它是否可用于 torch.cuda.is_available(),返回 true。

最奇怪的是,这个错误直到我运行代码大约 1.5 分钟后才出现。你会认为如果它检测不到 GPU,它会更早通知我。

在使用完全相同的笔记本运行其他 Pytorch 应用程序时,我在使用 Colab GPU 时没有遇到任何问题。我只能想象这个特定代码有问题,但返回的错误太奇怪了,我不得不在 StackOverflow 上询问以确保。

0 个答案:

没有答案