mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-10 01:02:56 +08:00
18 lines
286 B
Docker
18 lines
286 B
Docker
FROM python:3.10-slim
|
|
|
|
|
|
WORKDIR /opt/comfy_ui
|
|
|
|
|
|
RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 \
|
|
torch \
|
|
torchvision \
|
|
torchaudio
|
|
|
|
COPY . /opt/comfy_ui
|
|
RUN pip install -r requirements.txt
|
|
|
|
|
|
ENV CLI_ARGS=""
|
|
CMD python main.py ${CLI_ARGS}
|