pip is successfully installing h2o4gpu, still getting 'ImportError: No module named h2o4gpu'

pip is successfully installing h2o4gpu, still getting 'ImportError: No module named h2o4gpu'

我正在尝试按照此处列出的说明安装 h2o4gpu Python 模块:https://github.com/h2oai/h2o4gpu/issues/464

pip 表示已成功安装所有软件包,包括 h2o4gpu-0.1.0。

然而我仍然得到

    import h2o4gpu
ImportError: No module named h2o4gpu

kevin@Ubuntu-XPS:~/Downloads$ pip show tensorflow-gpu
Name: tensorflow-gpu
Version: 1.9.0
Summary: TensorFlow is an open source machine learning framework for everyone.
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: opensource@google.com
License: Apache 2.0
Location: /usr/local/lib/python2.7/dist-packages
Requires: grpcio, mock, protobuf, enum34, gast, wheel, absl-py, backports.weakref, termcolor, six, numpy, tensorboard, setuptools, astor
kevin@Ubuntu-XPS:~/Downloads$ pip show h2o4gpu
kevin@Ubuntu-XPS:~/Downloads$ 

因此显示 pip 确认安装了 tensorflow-gpu,但没有安装 h2o4gpu

我是 运行 Ubuntu 18.04 - 这可能是版本 18 尚不支持 h2o4gpu 的原因吗? import h2o 工作正常。

您正在混合 python2 和 python3。当 运行 pippython 都是 python2.7 时使用的内容(另请参阅 pip show tensorflow 的输出,其中它指的是 /usr/local/lib/python2.7/dist-packages).

您尝试使用的库只有 .whl for python 3.6(注意 .whl 文件名中的 py36

因此,您需要:

  • 切换到使用 pip3python3
  • 查找另一个适用于 python 2.7
  • 的库

我们不运送 python H2O4GPU 的 2.7 轮子,因此您需要按照 FlyingTeller 的建议使用 pip3。

最好使用 virtualenv 创建一个 python 环境以对系统造成最小的改变 python。

sudo apt-get install python3-pip
sudo pip3 install virtualenv
virtualenv -p python36 h2o4gpuenv
. h2o4gpuenv/bin/activate
pip install h2o4gpu-*.whl

现在在同一个虚拟环境中,启动 python 并尝试 import h2o4gpu