diff --git a/docker/dockerfile_cpp b/docker/dockerfile_cpp index 3febd80..896604b 100755 --- a/docker/dockerfile_cpp +++ b/docker/dockerfile_cpp @@ -36,7 +36,7 @@ COPY requirements.txt /tmp/requirements.txt # Install modelopt manually only if you need quantization tooling. RUN conda activate my &&\ uv pip install -r /tmp/requirements.txt &&\ - uv pip install onnxruntime-gpu onnx onnxscript pycuda cuda-python tensorrt-cu12 tensorrt-lean-cu12 tensorrt-dispatch-cu12 && nvidia-modelopt[torch] &&\ + uv pip install onnxruntime-gpu onnx onnxscript pycuda cuda-python tensorrt-cu12 tensorrt-lean-cu12 tensorrt-dispatch-cu12 &&\ conda install -y -c anaconda h5py &&\ conda install -y -c conda-forge libstdcxx-ng