运行 代码在 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 的卡的机器
我在 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 的卡的机器