From 747e02d60d6cd253f7b5667818b8ea9ed9fd3182 Mon Sep 17 00:00:00 2001
From: rumichi <rumichi@latestal.com>
Date: Sun, 23 Mar 2025 22:18:08 +0900
Subject: [PATCH] [docker] upgrade to torch 2.6 (#7442)

---
 docker/docker-cuda/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/docker-cuda/Dockerfile b/docker/docker-cuda/Dockerfile
index 15ac6ec2..aac9635e 100644
--- a/docker/docker-cuda/Dockerfile
+++ b/docker/docker-cuda/Dockerfile
@@ -1,6 +1,6 @@
-# Default use the NVIDIA official image with PyTorch 2.3.0
+# Default use the NVIDIA official image with PyTorch 2.6.0
 # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.12-py3
 FROM ${BASE_IMAGE}
 
 # Define environments