20 lines
651 B
Docker
20 lines
651 B
Docker
# 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"]
|
|
|