Missing 'nvgpu' package

[StrangeParadise]

Due to a recent PR of the referenced image 'pytorch-inference:1.12.0-gpu-py38'[1] that is used in the 'blogs/rag/data_ingestion_to_vectordb.ipynb' is missing the 'nvgpu' package which customer will run into 'ModuleNotFoundError: No module named 'nvgpu''. Please update the samples accordingly to address this issue.

Reference: [1] https://github.com/aws/deep-learning-containers/pull/2975

[wilburliu1123]

Faced same issue when going through this tutorial

[blackpearl1022]

Same.

But if you install nvgpu manually using this method pip install nvgpu or add nvgpu with specific version on the dependencies or requirements.txt, you will get this run.