如何在我的图形处理器而不是CPU上运行ImageAI

Question

所以我是个新手，我试着在谷歌上寻找答案，但它并不起作用。所以我正在尝试运行这个库ImageAI library

我可以在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上运行。您需要做的就是确保您安装了Tensorflow的GPU版本。