我如何 运行 ImageAI 在我的 GPU 而不是我的 CPU

Question

所以我对此很陌生，我试图在 google 上找到答案，但它有点不起作用。所以我正在尝试 运行 这个库 ImageAI library

我可以在 CPU 上正常地 运行 至少我认为 CPU 上 运行 只需调用 python test.py。我说得对吗

但由于模型预测需要很长时间，我想 运行 在我的 GPU 上进行。我想做的是创建一个 conda 环境并激活它，但在我这样做之后我得到了这个错误。

ModuleNotFoundError: No module named 'imageai.Classification'

虽然我的环境里安装了imageai

pip freeze | findstr imageai
imageai==2.1.5

正如你执行这条命令所见。我在这里做错了什么？

Answer 1

我找到了不需要 conda 环境的解决方案。如果可用，ImageAI 会自动在 GPU 上运行。您需要做的就是确保安装了 Tensorflow 的 GPU 版本。