Google Colab + Pytorch：RuntimeError：没有可用的 CUDA GPU

Question

错误截图：

您好，我正在尝试运行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 上询问以确认。

Answer 1

最近我遇到了类似的问题，其中 Cobal print(torch.cuda.is_available()) 为 True，但 print(torch.cuda.is_available()) 在特定项目中为 False。我们两个项目的代码都类似于 os.environ["CUDA_VISIBLE_DEVICES"]。可以用这段代码注释，发现GPU可以用了。

-------我的英语很差，我用谷歌翻译