在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配置。 選擇“最新可用”解決了該問題。