diff --git a/Dockerfile b/Dockerfile index b5cba4c..deeab58 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,7 +7,7 @@ RUN apt update && apt install -y --no-install-recommends libjpeg-dev python3-dev 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 --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