[BUG]No module named 'hierarchical_parameter_server' while I run docker with nvcr.io/nvidia/merlin/merlin-hugectr:24.06 image #468
Open
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
Labels
No labels