version bump to 118

AuScalableDroneCloud · Dec 14, 2023 · 9afa893 · 9afa893
1 parent a3e1ad8
commit 9afa893
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ml/Dockerfile b/ml/Dockerfile
@@ -5,9 +5,7 @@ LABEL maintainer "ASDC <[email protected]>"
 
 USER ${NB_UID}
 
-ARG TORCH_CUDA=117
+ARG TORCH_CUDA=118
 
 # Install torch 
-RUN pip install --quiet --no-cache-dir --pre \
-        torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu${TORCH_CUDA}
-
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA}