docker: bump PT to 2.4

Lightning-AI · Jul 24, 2024 · 1e23698 · 1e23698
1 parent 304932e
commit 1e23698
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_dockers/ubuntu-cuda/Dockerfile b/_dockers/ubuntu-cuda/Dockerfile
@@ -19,7 +19,7 @@ ARG CUDA_VERSION=11.8.0
 FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
 
 ARG PYTHON_VERSION=3.10
-ARG PYTORCH_VERSION=2.3
+ARG PYTORCH_VERSION=2.4
 
 SHELL ["/bin/bash", "-c"]
 # https://techoverflow.net/2019/05/18/how-to-fix-configuring-tzdata-interactive-input-when-building-docker-images/