use cuda 12.9.1 and add python 3.12 to base images (#3367)

This commit is contained in:
Wing Lian
2026-01-21 13:34:14 -05:00
committed by GitHub
parent 8ab9d9ea88
commit 8cd75cff9f
2 changed files with 33 additions and 1 deletions

View File

@@ -54,7 +54,7 @@ jobs:
strategy:
fail-fast: false
matrix:
python_version: ["3.11"]
python_version: ["3.11", "3.12"]
pytorch_version: ["2.8.0", "2.9.0", "2.9.1"]
timeout-minutes: 20