Update Dockerfile.local

awslabs · Jun 24, 2024 · 47a03e2 · 47a03e2
1 parent e8dbe65
commit 47a03e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local
@@ -44,7 +44,7 @@ RUN pip install \
 ARG DGL_VERSION=2.2.1
 ARG DGL_CUDA_VERSION=121
 ARG OGB_VERSION=1.3.6
-ARG TORCH_VERSION=2.1.2
+ARG TORCH_VERSION=2.3
 ARG TRANSFORMERS_VERSION=4.28.1
 
 FROM base as base-cpu