correct pip installation error in image build
This commit is contained in:
parent
26cc1072dd
commit
5b2ddeccdb
1 changed files with 4 additions and 1 deletions
|
|
@ -13,13 +13,16 @@ RUN apt-get update && \
|
|||
apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-venv \
|
||||
ffmpeg \
|
||||
git \
|
||||
build-essential \
|
||||
python3-dev && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu129
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu129
|
||||
|
||||
COPY . .
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue