【发布时间】:2019-08-20 21:40:22
【问题描述】:
我为完成的 Kaggle 竞赛创建了一个内核,在该竞赛中我使用了 pytorch。检查 cuda 是否可用时,它返回 False。
我从设置中检查了 GPU 选项,它说它在底部栏中带有资源信息。我试图在没有任何更改的情况下重新启动会话。 可能是什么问题呢? (可能只安装了 pytorch 的 cpu?)
【问题讨论】:
标签: jupyter-notebook pytorch kaggle
我为完成的 Kaggle 竞赛创建了一个内核,在该竞赛中我使用了 pytorch。检查 cuda 是否可用时,它返回 False。
我从设置中检查了 GPU 选项,它说它在底部栏中带有资源信息。我试图在没有任何更改的情况下重新启动会话。 可能是什么问题呢? (可能只安装了 pytorch 的 cpu?)
【问题讨论】:
标签: jupyter-notebook pytorch kaggle
问题出在设置中选择的 docker 配置。选择“最新可用”解决了这个问题。
【讨论】:
在右侧控制面板的设置选项中启动内核和turn-on GPU。然后运行以下命令。
[1]:import torch
[2]:torch.cuda.device_count()
Out[2]:1
[3]:torch.cuda.is_available()
Out[3]:True
【讨论】: