运行 代码在 GPU 上而不是 CPU with detecto

Run code on GPU instead of CPU with detecto

我在 Python 中使用机器学习和 detecto。但是,每当我 运行 它时,我都会收到一条警告说

It looks like you're training your model on a CPU. Consider switching to a GPU; otherwise, 
this method can take hours upon hours or even days to finish. For more information, see 
https://detecto.readthedocs.io/en/latest/usage/quickstart.html#technical-requirements 

我有一个 Intel(R) 高清显卡 4600 形式的 GPU,但由于某种原因,代码 运行ning 在 CPU.我已经检查了它给出的 link 上面写着

By default, Detecto will run all heavy-duty code on the GPU if it’s available and on the CPU otherwise.

如果计算机没有可以使用的 GPU,建议使用 Google Collab,但我有一个,不想使用 Google Collab。

为什么 运行ning 在 CPU 而不是 GPU 上?我该如何解决?我收到警告的代码部分是

losses = fitmodel(loader, Test_dataset, epochs=25, lr_step_size=5, 
learning_rate=0.001, verbose=True)

该代码确实有效,但是 运行 需要很长时间,因此希望能够 运行 在 GPU 上执行它以节省时间。

detecto 所指的 GPU 需要是支持 CUDA 的 Nvidia GPU。所以你的 Intel(R) HD 显卡 4600 不符合这个标准。

Detecto 在内部使用 pytorch,GPU 支持 is based on CUDA。因此,为了使用 GPU,您需要移动到具有支持 CUDA 的卡的机器