add torchaudio into container
Docker Build and Release Upload / build-rocm (push) Failing after 1m33s

This commit is contained in:
2026-05-17 20:26:39 -07:00
parent f58b4fa820
commit 3034311730
+21 -8
View File
@@ -1,8 +1,25 @@
# Get cuda version from https://hub.docker.com/r/nvidia/cuda
FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
# Install system dependencies
RUN pip install imageio controlnet_aux
# Set ENV
ENV USE_CUDA=1
ENV BUILD_SOX=1
# Install system packages to build torchaudio
RUN apt update && apt install -y \
libavformat-dev \
libavcodec-dev \
libavutil-dev \
libavdevice-dev \
libavfilter-dev \
libswresample-dev \
sox \
libsox-dev
# 2. Clone and build the torchaudio
RUN git clone -b main https://github.com/pytorch/audio.git && \
cd audio && \
pip install --no-build-isolation --no-deps -v .
# Set working directory
WORKDIR /app
@@ -10,13 +27,9 @@ WORKDIR /app
# Clone ComfyUI
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
# Link in python
RUN ln -sf /usr/bin/python3 /usr/bin/python && \
ln -sf /usr/bin/pip3 /usr/bin/pip
# Install
# Install depencies
RUN grep -vE "^(torch|#)" requirements.txt | xargs pip install && \
pip install torchsde
pip install torchsde imageio controlnet_aux
# Set custom Nodes
WORKDIR /app/custom_nodes