chore(ci): use nightly CPU PyTorch for Docker image

kouprlabs · Jun 13, 2024 · 16ff0a4 · 16ff0a4
1 parent aa7377f
commit 16ff0a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/language/Dockerfile b/language/Dockerfile
@@ -10,7 +10,7 @@ RUN apt-get install -y curl build-essential python3-poetry rust-all
 
 RUN poetry install --no-root
 
-RUN pip3 install torch==2.3.1
+RUN pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 
 ENV PIP_DEFAULT_TIMEOUT=900