From 3034311730eaea7ac80a6b5a19a71a1125d17cfb Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Sun, 17 May 2026 20:26:39 -0700
Subject: [PATCH] add torchaudio into container

---
 comfyui-cuda-dockerfile | 29 +++++++++++++++++++++--------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/comfyui-cuda-dockerfile b/comfyui-cuda-dockerfile
index 49340f6..402d72f 100644
--- a/comfyui-cuda-dockerfile
+++ b/comfyui-cuda-dockerfile
@@ -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