[devops] update torch versoin in ci (#5217)

This commit is contained in:
Hongxin Liu
2024-01-03 11:46:33 +08:00
committed by GitHub
parent d799a3088f
commit 7f3400b560
7 changed files with 11 additions and 11 deletions

View File

@@ -34,7 +34,7 @@ jobs:
fail-fast: false
matrix: ${{fromJson(needs.matrix_preparation.outputs.matrix)}}
container:
image: hpcaitech/pytorch-cuda:1.12.0-11.3.0
image: hpcaitech/pytorch-cuda:2.0.0-11.7.0
timeout-minutes: 10
steps:
- name: 📚 Checkout