Skip to content

[BUG]No module named 'hierarchical_parameter_server' while I run docker with nvcr.io/nvidia/merlin/merlin-hugectr:24.06 image #468

Open
@HuuY

Description

Describe the bug
In order to run the demo fellow this guide:
https://github.com/NVIDIA-Merlin/HugeCTR/blob/main/hps_tf/notebooks/hierarchical_parameter_server_demo.ipynb
I run docker with image nvcr.io/nvidia/merlin/merlin-hugectr:24.06 with:
$ docker run --gpus=all --name huge_ctr_hugo -it --cap-add SYS_NICE -v /home/:/home/ -w /home/hugo -it --privileged nvcr.io/nvidia/merlin/merlin-hugectr:24.06
But when I run $ python3 -c "import hierarchical_parameter_server as hps"
error occurs like No module named 'hierarchical_parameter_server'

So is there something I did wrong or the api changes?
Thanks in advance.

Environment (please complete the following information):

  • OS: [e.g. Ubuntu 20.04]
  • Graphic card: [e.g. 8 cards nvidia L20]
  • CUDA version: [e.g. CUDA 12.6]
  • Docker image :nvcr.io/nvidia/merlin/merlin-hugectr:24.06

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions