Update Dockerfile to install build-essential and update PyTorch version
This commit is contained in:
parent
8e056cbdf2
commit
e4140cd299
1 changed files with 4 additions and 2 deletions
|
|
@ -21,10 +21,12 @@ RUN apt-get update && \
|
|||
python3 \
|
||||
python3-pip \
|
||||
ffmpeg \
|
||||
git && \
|
||||
git \
|
||||
build-essential \
|
||||
python3-dev && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
||||
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
||||
|
||||
COPY . .
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue