Fix container image for rocm

This commit is contained in:
doctorpangloss 2025-09-23 15:33:23 -07:00
parent 1d3dd20b67
commit 74d77a3757

View File

@ -64,7 +64,7 @@ jobs:
runner:
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
friendly_name: "Python 3.12 ROCm 7.0 Torch 2.7.1"
container: "rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1"
container: "rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1"
steps:
- run: |
apt-get update