问题描述
我正在尝试安装 plaidml-keras,以便我可以在 python/jupyter 中使用带有 Keras 的非 Nvidia GPU。在清除了其他几个障碍后,我达到了:
plaidbench keras mobilenet
但它错误两次
ImportError: cannot import name 'object_list_uid' from 'keras.utils.generic_utils' (/Users/me/sprinthive/src/notebooks/nbenv/lib/python3.7/site-packages/keras/utils/generic_utils.py)
File "/Users/me/sprinthive/src/notebooks/nbenv/lib/python3.7/site-packages/plaidbench/frontend_keras.py",line 321,in __init__
raise core.ExtrasNeeded(['plaidml-keras'])
plaidbench.core.ExtrasNeeded: Missing needed packages for benchmark; to fix,pip install plaidml-keras
尽管已经安装了 plaidml-keras:
pip freeze | grep plaid
plaidbench==0.6.4
plaidml==0.6.4
plaidml-keras==0.6.4
[我使用 0.6.4 使其在 macOS 10.13 High Sierra 上运行]
谢谢!
解决方法
我通过使用 Anaconda 创建虚拟环境来解决这个问题。请注意,我在 Windows 上工作,所以这可能不是您问题的解决方案。如果我不得不猜测,在导致 python 包问题之前我安装了一些东西。我认为这与 tensorflow 库有关,但我还没有深入研究。我建议您在 Mac 上尝试一个全新的虚拟环境,在其中安装 plaidml 包。之前的错误信息完全一样。