在 Kaggle 内核中，在选中 torch.cuda.is_available() 时选择了 GPU 选项，它说不可用

Question

我为完成的Kaggle竞赛创建了一个内核，其中我使用了pytorch。检查 cuda 是否可用时，它会返回 False。

我从设置中检查了 GPU 选项，它说它在底部栏中打开，其中包含资源信息。我尝试重新启动会话而不进行任何更改。可能是什么问题？(也许只安装了 CPU 的 PyTorch？

Answer 1

启动内核并在右侧控制面板上的设置选项中turn-on GPU。然后运行以下命令。

    [1]:import torch
    [2]:torch.cuda.device_count()
    Out[2]:1
    [3]:torch.cuda.is_available()
    Out[3]:True

Answer 2

问题出在设置中选择的 docker 配置上。选择"最新可用"解决了问题。