From 0bf38c1cd4c553483308ea6698f9370f7b8e8b60 Mon Sep 17 00:00:00 2001
From: albe873 <93786475+albe873@users.noreply.github.com>
Date: Fri, 13 Dec 2024 23:14:10 +0100
Subject: [PATCH] changed base image

---
 Dockerfile | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index e8f8837..948f9aa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,17 @@
-FROM rocm/pytorch:rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1
+# https://rocm.docs.amd.com/projects/install-on-linux/en/develop/install/3rd-party/pytorch-install.html
 
-RUN pip install -U pip && pip install wyoming==1.2.0 openai-whisper==20231117 tokenizers==0.13.*
+FROM rocm/dev-ubuntu-24.04
+
+RUN apt update && apt install -y libjpeg-dev python3-dev python3-pip 
+
+RUN pip3 install --break-system-packages wheel setuptools
+RUN pip3 install --break-system-packages --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2/
+
+RUN pip3 install --break-system-packages wyoming openai-whisper tokenizers
 
 COPY src /src
 
 WORKDIR /src
 
 ENTRYPOINT ["/src/run.sh"]
+