CI: Try not even pinning torch

ml-energy · Jul 7, 2023 · eab686b · eab686b
1 parent ee682fc
commit eab686b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -42,7 +42,7 @@ RUN if [ "$TARGETARCH" = "amd64" ]; then \
     && unset CONDA_INSTALLER_PATH
 
 # Install PyTorch and CUDA Toolkit
-RUN pip install torch==2.0.1 torchvision --index-url https://download.pytorch.org/whl/cu118
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
 
 # Place stuff under /workspace
 WORKDIR /workspace