# https://rocm.docs.amd.com/projects/install-on-linux/en/develop/install/3rd-party/pytorch-install.html

FROM rocm/dev-ubuntu-24.04

RUN apt update && apt install -y --no-install-recommends libjpeg-dev python3-dev python3-pip && \
	apt clean && \
	rm -r /var/lib/apt/lists/*

RUN pip3 install --break-system-packages --no-cache-dir wheel setuptools
RUN pip3 install --break-system-packages --no-cache-dir --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2/

RUN pip3 install --break-system-packages --no-cache-dir wyoming openai-whisper tokenizers

COPY src /src

WORKDIR /src

ENTRYPOINT ["/src/run.sh"]