From 137d84d1b408e70e679fc3604109298e76c9bd63 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 13 Jul 2024 09:41:51 -0400
Subject: [PATCH] add torch 2.3.1 base image (#1745)

---
 .github/workflows/base.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index d215ea44c..f8eaff270 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -37,6 +37,11 @@ jobs:
             python_version: "3.11"
             pytorch: 2.3.0
             torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
+          - cuda: "121"
+            cuda_version: 12.1.0
+            python_version: "3.11"
+            pytorch: 2.3.1
+            torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
     steps:
       - name: Checkout
         uses: actions/checkout@v3