diff --git a/dockerfiles/csci4150/s25/Dockerfile b/dockerfiles/csci4150/s25/Dockerfile
index 526c959..67f0c97 100644
--- a/dockerfiles/csci4150/s25/Dockerfile
+++ b/dockerfiles/csci4150/s25/Dockerfile
@@ -4,4 +4,6 @@ RUN apt-get update
 
 RUN apt-get install python3.10 pip -y
 
-RUN pip install torch==2.6.0 numpy==2.2.4 matplotlib==3.10.1
\ No newline at end of file
+RUN pip install numpy==2.2.4 matplotlib==3.10.1
+
+RUN pip install torch==2.6.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu