scott/rocm-wyoming-whisper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed base image

Browse Source
This commit is contained in:
albe873
2024-12-13 23:14:10 +01:00
parent dddce51ec8
commit 0bf38c1cd4
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
Dockerfile
View File

@@ -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 COPY src /src
WORKDIR /src WORKDIR /src
ENTRYPOINT ["/src/run.sh"] ENTRYPOINT ["/src/run.sh"]