【问题标题】:In a Kaggle kernel while having selected the GPU option when checking torch.cuda.is_available(), it says is not available在 Kaggle 内核中,在检查 torch.cuda.is_available() 时选择了 GPU 选项,它说不可用
【发布时间】:2019-08-20 21:40:22
【问题描述】:

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

我从设置中检查了 GPU 选项,它说它在底部栏中带有资源信息。我试图在没有任何更改的情况下重新启动会话。 可能是什么问题呢? (可能只安装了 pytorch 的 cpu?)

【问题讨论】:

    标签: jupyter-notebook pytorch kaggle


    【解决方案1】:

    问题出在设置中选择的 docker 配置。选择“最新可用”解决了这个问题。

    【讨论】:

      【解决方案2】:

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

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

      【讨论】:

        猜你喜欢
        • 2020-07-19
        • 1970-01-01
        • 1970-01-01
        • 2014-03-28
        • 1970-01-01
        • 1970-01-01
        • 2013-03-17
        • 1970-01-01
        • 1970-01-01
        相关资源
        最近更新 更多