Skip to content

Commit d50f745

Browse files
fix: 更新Dockerfile中PyTorch版本为rocm6.2
1 parent d425c0d commit d50f745

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

.github/workflows/docker-build.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
name: Build and Push Docker Image to GitHub Container Registry
22

33
on:
4+
push:
5+
branches:
6+
- main
47
workflow_dispatch:
58

69
jobs:

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git /ComfyUI
1515

1616
# 安装 PyTorch,指定 ROCm 相关包
1717
RUN pip3 install --no-cache-dir torch torchvision torchaudio \
18-
--index-url https://download.pytorch.org/whl/rocm6.3
18+
--index-url https://download.pytorch.org/whl/rocm6.2
1919

2020
# 安装 ComfyUI 项目依赖
2121
WORKDIR /ComfyUI

0 commit comments

Comments
 (0)