rocm 6.1.2 seems to have a problem, restore rocm 5.4.2

dockerfile-distrib

@@ -43,10 +43,10 @@ RUN apt-get install -y \
     python3-scipy
 
 RUN curl -fsSL https://repo.radeon.com/rocm/rocm.gpg.key | sudo gpg --dearmor -o /etc/apt/trusted.gpg.d/rocm-keyring.gpg
-RUN echo 'deb [arch=amd64 signed-by=/etc/apt/trusted.gpg.d/rocm-keyring.gpg] https://repo.radeon.com/rocm/apt/6.1.2 jammy main' | sudo tee /etc/apt/sources.list.d/rocm.list
+RUN echo 'deb [arch=amd64 signed-by=/etc/apt/trusted.gpg.d/rocm-keyring.gpg] https://repo.radeon.com/rocm/apt/5.4.2 jammy main' | sudo tee /etc/apt/sources.list.d/rocm.list
 RUN echo -e 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600
 RUN apt-get update
-RUN apt-get install -y rocm-hip-sdk
+RUN apt-get install -y rocm-hip-sdk && rm -rf /var/lib/apt/lists/*
 
 RUN apt-get autoremove -y
 RUN apt-get autoclean -y