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 \
|
||||||
python3-pip \
|
python3-pip \
|
||||||
ffmpeg \
|
ffmpeg \
|
||||||
git && \
|
git \
|
||||||
|
build-essential \
|
||||||
|
python3-dev && \
|
||||||
rm -rf /var/lib/apt/lists/*
|
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 . .
|
COPY . .
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue