在Kaggle内核中，当检查torch.cuda.is_available（）时选择了GPU选项时，它说不可用

Question

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

我从设置中检查了GPU选项，它说它在底部的资源信息栏中。我尝试重新启动会话，没有进行任何更改。 可能是什么问题呢？ （也许只安装了pytorch cpu？）

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配置。选择“最新可用”解决了该问题。