18 lines
534 B
Docker
18 lines
534 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 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"]
|
|
|