From eab686b709461f90ba91bad14869e92fef20eae3 Mon Sep 17 00:00:00 2001
From: Jae-Won Chung <jwnchung@umich.edu>
Date: Fri, 7 Jul 2023 18:12:35 -0400
Subject: [PATCH] CI: Try not even pinning torch

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 384f999b..755309cc 100644
--- 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