diff --git a/Dockerfile.ci b/Dockerfile.ci index 173d1c62e5ff..384aac5e1481 100644 --- a/Dockerfile.ci +++ b/Dockerfile.ci @@ -37,7 +37,7 @@ RUN pip install nemo_run@git+https://github.com/NVIDIA/NeMo-Run.git@${NEMU_RUN_T # Install NeMo requirements ARG TE_TAG=7d576ed25266a17a7b651f2c12e8498f67e0baea -ARG MODELOPT_VERSION=0.15.0 +ARG MODELOPT_VERSION=0.17.0 ARG MCORE_TAG=core_r0.9.0 ARG APEX_TAG=810ffae374a2b9cb4b5c5e28eaeca7d7998fca0c RUN \ diff --git a/docs/source/starthere/intro.rst b/docs/source/starthere/intro.rst index 435e3e24350e..c3c3304082dc 100644 --- a/docs/source/starthere/intro.rst +++ b/docs/source/starthere/intro.rst @@ -102,7 +102,7 @@ This final step involves installing the TensorRT Model Optimizer package. .. code-block:: bash - pip install nvidia-modelopt[torch]~=0.15.0 --extra-index-url https://pypi.nvidia.com + pip install nvidia-modelopt[torch]~=0.17.0 --extra-index-url https://pypi.nvidia.com .. code-block:: bash