Fix URL per feedback

This commit is contained in:
Njal Kuhn 2024-08-26 18:07:42 +02:00 committed by GitHub
parent d046116084
commit 1ff0e8ac07
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,7 +1,7 @@
FROM python:3.12.5-bookworm FROM python:3.12.5-bookworm
RUN pip install jinja2 numpy RUN pip install jinja2 numpy
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
#RUN pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 #RUN pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
RUN apt update -y RUN apt update -y
RUN apt install -y git RUN apt install -y git