Hi,

I have been using a base Docker image on 7900xtx with WSL:

FROM rocm/pytorch:rocm6.3.1_ubuntu22.04_py3.10_pytorch

RUN useradd -m -s /bin/bash jupyter_user && \
    mkdir -p /workspace/node_modules && \
    chown -R jupyter_user:jupyter_user /workspace && \
    chmod -R 755 /workspace && \
    apt-get update && \
    apt-get install -y \
    ffmpeg \
    git \
    curl \
    unzip && \
    rm -rf /var/lib/apt/lists/*

WORKDIR /workspace

CMD ["/bin/bash"]

This setup works, and I can confirm it with:

import torch
torch.cuda.is_available()

However, as soon as I install torchaudio, it seems to start downloading a new version of torch, which messes things up.

I found this page but I'm unsure which .whl file to try: https://download.pytorch.org/whl/torchaudio/

Also, WhisperX seems to have other issues on ROCm: https://github.com/m-bain/whisperX/issues/566

Can anyone clarify which popular libraries like this still don't work properly on ROCm?

So there wasn't a solution out that would allow you to submit your audio/video files to a dockerized flask app and have your subtitle files etc generated as if you were using whisper via the command line... so I made one. Go check it out here jlonge4/whisperAI-flask-docker: I built this project because there was no user friendly way to upload a file to a dockerized flask web form and have whisper do its thing via CLI in the background. Now there is. Enjoy! (github.com)