错误截图:
您好,我正在尝试运行 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 上询问以确保。