【问题标题】:How can I run ImageAI on my GPU and not my CPU如何在我的 GPU 而不是我的 CPU 上运行 ImageAI
【发布时间】:2021-11-26 15:47:10
【问题描述】:

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

我可以在 CPU 上正常运行它,至少我认为它通过调用 python test.py 在 CPU 上运行。我说的对吗

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

ModuleNotFoundError: No module named 'imageai.Classification'

虽然我的环境中安装了 imageai

pip freeze | findstr imageai
imageai==2.1.5

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

【问题讨论】:

  • 还有其他几篇文章here
  • @RobertHarvey 是的,但这不是同一个问题。一个人根本不认识图书馆,但不认识模块
  • 我认为无论如何你都应该尝试遵循这些文章中的建议。他们解释了如何正确使用 pip。包管理器的存在是因为它们有助于防止像您这样的依赖问题。
  • @RobertHarvey 将更仔细地查看我已经做过的文章,但也许​​我错过了一些东西。但据我所知,我安装了这个库的所有要求

标签: tensorflow conda image-recognition custom-training imageai


【解决方案1】:

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2020-01-13
    • 2020-07-18
    • 2020-07-26
    • 2016-02-08
    • 1970-01-01
    • 2019-10-23
    • 2011-10-02
    • 2015-02-24
    相关资源
    最近更新 更多