minor fix

rocm-base-aarch64-dockerfile

@@ -34,6 +34,11 @@ RUN apt update -y && \
     zlib1g-dev \
     libncurses5-dev \
     libopenblas-dev \
+    libjpeg-dev \
+    libpng-dev \
+    libavcodec-dev \
+    libavformat-dev \
+    libswscale-dev \
     libpthreadpool-dev \
     libgoogle-glog-dev \
     libgtest-dev \
@@ -147,11 +152,21 @@ RUN git clone https://github.com/pytorch/pytorch.git && \
 # Build pytorch-audio
 FROM pytorch AS torchaudio
 WORKDIR /build/torchaudio
+RUN wget https://github.com/pytorch/audio/archive/refs/tags/v2.11.0.tar.gz && \
+    tar -xf v2.11.0.tar.gz && \
+    cd audio-2.11.0 && \
+    pip install --no-build-isolation --no-deps -v .
+    # Tested Working via comfyui-cuda-dockerfile
 
 # Build pytorch-vision
 FROM torchaudio AS torchvision
 WORKDIR /build/torchaudio
-RUN git clone https://github.com/pytorch/vision.git
+ENV FORCE_CUDA=0
+RUN git clone https://github.com/pytorch/vision.git . && \
+    pip install numpy pillow ninja setuptools wheel && \
+    python setup.py bdist_wheel && \
+    pip install dist/torchvision-*.whl
+    # Not tested and not verified ROCM build
 
 # Build final stripped down image
 FROM ubuntu:26.04