Skip to content

Commit c9c41db

Browse files
authored
Install CPU-only Torch (#32)
We couldn't find much about the error Submitty was giving us online, but ChatGPT said it had to do with trying to load CUDA which makes sense since we don't have access to it, so I've changed the environment to install the CPU-only version of torch. Hopefully, this fixes the error.
1 parent 3ae1924 commit c9c41db

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dockerfiles/csci4150/s25/Dockerfile

+3-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,6 @@ RUN apt-get update
44

55
RUN apt-get install python3.10 pip -y
66

7-
RUN pip install torch==2.6.0 numpy==2.2.4 matplotlib==3.10.1
7+
RUN pip install numpy==2.2.4 matplotlib==3.10.1
8+
9+
RUN pip install torch==2.6.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

0 commit comments

Comments
 (0)