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

Question

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

我能够在 CPU 上正常运行它，至少我认为它只需调用 python test.py 就可以在 CPU 上运行。我在那里正确吗

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

ModuleNotFoundError: No module named 'imageai.Classification'

虽然我在我的环境中安装了 imageai

pip freeze | findstr imageai
imageai==2.1.5

正如您通过执行此命令所看到的。我在这里做错了什么？

Answer 1

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