diff --git a/docker/Dockerfile b/docker/Dockerfile index 4885701..f34cd0f 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -15,7 +15,7 @@ COPY . . # Install package from source code, compile translations RUN ./venv/bin/pip install Babel==2.12.1 && ./venv/bin/python scripts/compile_locales.py \ - && ./venv/bin/pip install torch --extra-index-url https://download.pytorch.org/whl/cpu \ + && ./venv/bin/pip install torch==2.0.1 --extra-index-url https://download.pytorch.org/whl/cpu \ && ./venv/bin/pip install . \ && ./venv/bin/pip cache purge