diff --git a/docker/cuda.Dockerfile b/docker/cuda.Dockerfile index d35b85f..40641de 100644 --- a/docker/cuda.Dockerfile +++ b/docker/cuda.Dockerfile @@ -34,7 +34,7 @@ RUN if [ "$with_models" = "true" ]; then \ RUN pip3 install Babel==2.12.1 && python3 scripts/compile_locales.py \ && pip3 install "numpy<2" \ && pip3 install . \ - && pip3 uninstall onnxruntime && pip3 install --no-cache-dir onnxruntime-gpu>=1.10.0 \ + && pip3 uninstall -y onnxruntime && pip3 install --no-cache-dir onnxruntime-gpu>=1.10.0 \ && pip3 cache purge # Depending on your cuda install you may need to uncomment this line to allow the container to access the cuda libraries