Switch to supported ROCm image

This commit is contained in:
Benjamin Berman 2024-07-30 11:42:14 -07:00 committed by GitHub
parent a94cd0b626
commit c6ce11b421
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -53,7 +53,7 @@ jobs:
- labels: [self-hosted, Linux, X64, cpu]
container: "ubuntu:22.04"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4"
container: "rocm/pytorch:rocm6.1.3_ubuntu22.04_py3.10_pytorch_release-2.1.2"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]
container: "nvcr.io/nvidia/pytorch:24.03-py3"
steps: