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

Image size reduction

Browse Source
This commit is contained in:
albe873
2024-12-17 21:07:37 +01:00
parent 7e4b0e3932
commit 206c91d3ad
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Dockerfile
View File

@@ -2,12 +2,14 @@
FROM rocm/dev-ubuntu-24.04
RUN apt update && apt install -y libjpeg-dev python3-dev python3-pip
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 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 --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 wyoming openai-whisper tokenizers
RUN pip3 install --break-system-packages --no-cache-dir wyoming openai-whisper tokenizers
COPY src /src