From 3591bcfaf9793c8cb09e41d5d66b55a438042c06 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 31 Oct 2024 13:27:49 -0400
Subject: [PATCH] add torch 2.5.1 for base image (#2010)

---
 .github/workflows/base.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index c94093bc9..cd99b9772 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -40,7 +40,7 @@ jobs:
             cuda_version: 12.4.1
             cudnn_version: ""
             python_version: "3.11"
-            pytorch: 2.5.0
+            pytorch: 2.5.1
             torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
     steps:
       - name: Checkout