【发布时间】:2021-04-25 17:39:37
【问题描述】:
我在加速器选项卡中启用了 GPU 选项,但是当我尝试训练我的模型时,GPU 并没有处理我的数据,而是只有 CPU 正在处理。
【问题讨论】:
-
欢迎来到 Stack Overflow!请收下tour 并阅读How to Ask。
标签: image-processing neural-network conv-neural-network data-science kaggle
我在加速器选项卡中启用了 GPU 选项,但是当我尝试训练我的模型时,GPU 并没有处理我的数据,而是只有 CPU 正在处理。
【问题讨论】:
标签: image-processing neural-network conv-neural-network data-science kaggle
在您的训练代码中似乎存在问题,您可能无法正确地将模型和输入发送到设备,因此最终只能使用您的 CPU。查看这些 Kaggle 示例以了解如何使用:
【讨论】:
您需要专门使用 https://rapids.ai/ 之类的包将数据从 CPU 移动到 GPU。我假设您使用 pandas 之类的东西来进行数据操作。
【讨论】: