scott
937b6e1201
All checks were successful
Docker Build and Publish / Build Docker image amd publish to Gitea (push) Successful in 16m20s
20 lines
643 B
Docker
20 lines
643 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/rocm6.4/
|
|
|
|
RUN pip3 install --break-system-packages --no-cache-dir wyoming openai-whisper tokenizers
|
|
|
|
COPY src /src
|
|
|
|
WORKDIR /src
|
|
|
|
ENTRYPOINT ["/src/run.sh"]
|
|
|