Disable ROCm again

This commit is contained in:
Benjamin Berman 2024-11-10 20:33:56 -08:00 committed by GitHub
parent 772e768fe8
commit 97400a679b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -54,8 +54,8 @@ jobs:
container: "ubuntu:22.04"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]
container: "nvcr.io/nvidia/pytorch:24.03-py3"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0"
# - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
# container: "rocm/pytorch:rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0"
steps:
- run: |
apt update