add torchaudio into container
Docker Build and Release Upload / build-rocm (push) Failing after 1m33s
Docker Build and Release Upload / build-rocm (push) Failing after 1m33s
This commit is contained in:
+21
-8
@@ -1,8 +1,25 @@
|
|||||||
# Get cuda version from https://hub.docker.com/r/nvidia/cuda
|
# Get cuda version from https://hub.docker.com/r/nvidia/cuda
|
||||||
FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
|
FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
|
||||||
|
|
||||||
# Install system dependencies
|
# Set ENV
|
||||||
RUN pip install imageio controlnet_aux
|
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
|
# Set working directory
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
@@ -10,13 +27,9 @@ WORKDIR /app
|
|||||||
# Clone ComfyUI
|
# Clone ComfyUI
|
||||||
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
|
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
|
||||||
|
|
||||||
# Link in python
|
# Install depencies
|
||||||
RUN ln -sf /usr/bin/python3 /usr/bin/python && \
|
|
||||||
ln -sf /usr/bin/pip3 /usr/bin/pip
|
|
||||||
|
|
||||||
# Install
|
|
||||||
RUN grep -vE "^(torch|#)" requirements.txt | xargs pip install && \
|
RUN grep -vE "^(torch|#)" requirements.txt | xargs pip install && \
|
||||||
pip install torchsde
|
pip install torchsde imageio controlnet_aux
|
||||||
|
|
||||||
# Set custom Nodes
|
# Set custom Nodes
|
||||||
WORKDIR /app/custom_nodes
|
WORKDIR /app/custom_nodes
|
||||||
|
|||||||
Reference in New Issue
Block a user