diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 92192d5..4006dc8 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -32,11 +32,11 @@ FROM reqs AS torch-audio
 
 WORKDIR /builder
 
-RUN git clone --depth 1 https://github.com/pytorch/audio.git && \
+RUN git clone --depth 1 --branch v2.2.2 https://github.com/pytorch/audio.git && \
     cd audio && \
     git submodule update --init --recursive && \
     export PIP_BREAK_SYSTEM_PACKAGES=1 && \
-    CC=gcc CXX=g++ BUILD_ROCM=1 python3 setup.py install && \
+    USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
     cd .. && rm -rf audio
 
 FROM torch-audio AS runner