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

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。