wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Use updated containers for rocm to try to fix its crashes

Browse Source
This commit is contained in:
Benjamin Berman 2024-07-27 18:45:25 -07:00 committed by GitHub
parent 2c0d6f2ba9
commit 38312e3c9f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
.github/workflows/test.yml vendored
View File

@ -51,12 +51,11 @@ jobs:
matrix:
runner:
- labels: [self-hosted, Linux, X64, cpu]
container: "ubuntu"
# todo: re-enable these when they actually work lol
# - labels: [self-hosted, Linux, X64, ipex-a380-6gb]
# container: "intel/intel-optimized-pytorch:2.1.20-xpu-pip-base"
container: "ubuntu:22.04"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
container: "rocm/pytorch:rocm6.1.3_ubuntu22.04_py3.10_pytorch_release-2.1.2"
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]
container: "nvcr.io/nvidia/pytorch:24.03-py3"
steps: