diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 3a8384f..9338796 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -22,6 +22,9 @@ RUN apt update -y && apt install -y --no-install-recommends \
     python3-pip \
     python3-dev \
     python3-torch-rocm \
+    libhipblas-dev \
+    librocblas-dev \
+    hip-dev \
     git \
     libgl1 \
     libglib2.0-bin \
@@ -43,7 +46,7 @@ FROM reqs AS torch-audio
 
 WORKDIR /builder
 
-RUN git clone --depth 1 --branch v2.2.2 https://github.com/pytorch/audio.git && \
+RUN git clone --depth 1 --branch v2.9.1 https://github.com/pytorch/audio.git && \
     cd audio && \
     git submodule update --init --recursive && \
     USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \