Pin torch to 2.0.1

This commit is contained in:
Piero Toffanin 2023-10-29 21:40:23 -04:00
parent be8637e0c6
commit 7be612419b

View file

@ -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