diff --git a/Dockerfile b/Dockerfile
index 7b8bc1c8..a7e6700e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -27,7 +27,7 @@ WORKDIR /modded-nanogpt
 RUN python -m pip install --upgrade pip && \
     pip install -r requirements.txt
 
-RUN pip install --pre torch==2.6.0.dev20241203+cu126 --index-url https://download.pytorch.org/whl/nightly/cu126 --upgrade
+RUN pip install --pre torch==2.6.0.dev20241231+cu126 --index-url https://download.pytorch.org/whl/nightly/cu126 --upgrade
 
 CMD ["bash"]
 ENTRYPOINT []