From 7be612419b7eae31f336ed98ad19a6813000029d Mon Sep 17 00:00:00 2001 From: Piero Toffanin Date: Sun, 29 Oct 2023 21:40:23 -0400 Subject: [PATCH] Pin torch to 2.0.1 --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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