feat(docker): Add TensorRT modelopt

Author: nkemnitz

docker/pytorch_base/Dockerfile.prebuilt

@@ -33,6 +33,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     && /opt/conda/bin/conda clean -ya \
     && /opt/conda/bin/pip install torch==${TORCH_VERSION} torchvision==${TORCHVISION_VERSION} \
       tensorrt torch-tensorrt==${TORCH_TENSORRT_VERSION} --extra-index-url https://download.pytorch.org/whl/cu$(echo $CUDA_VERSION | tr -d .) \
+    && /opt/conda/bin/pip install "nvidia-modelopt[all]" --extra-index-url https://pypi.nvidia.com \
     # && /opt/conda/bin/pip install tensorrt torch-tensorrt==${TORCH_TENSORRT_VERSION} -f https://github.com/NVIDIA/Torch-TensorRT/releases \
     && apt-mark auto '.*' > /dev/null \
     && apt-mark manual $preexistingAptMark ca-certificates > /dev/null \