Skip to content

Commit dc224c4

Browse files
sayakpaulDisty0
authored andcommitted
[Docker] include python3.10 dev and solve header missing problem (huggingface#8865)
include python3.10 dev and solve header missing problem
1 parent 1bbf505 commit dc224c4

File tree

4 files changed

+10
-7
lines changed

4 files changed

+10
-7
lines changed

docker/diffusers-pytorch-compile-cuda/Dockerfile

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,24 +16,24 @@ RUN apt install -y bash \
1616
ca-certificates \
1717
libsndfile1-dev \
1818
libgl1 \
19-
python3.9 \
20-
python3.9-dev \
19+
python3.10 \
20+
python3.10-dev \
2121
python3-pip \
22-
python3.9-venv && \
22+
python3.10-venv && \
2323
rm -rf /var/lib/apt/lists
2424

2525
# make sure to use venv
26-
RUN python3.9 -m venv /opt/venv
26+
RUN python3.10 -m venv /opt/venv
2727
ENV PATH="/opt/venv/bin:$PATH"
2828

2929
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
30-
RUN python3.9 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
31-
python3.9 -m uv pip install --no-cache-dir \
30+
RUN python3.10 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
31+
python3.10 -m uv pip install --no-cache-dir \
3232
torch \
3333
torchvision \
3434
torchaudio \
3535
invisible_watermark && \
36-
python3.9 -m pip install --no-cache-dir \
36+
python3.10 -m pip install --no-cache-dir \
3737
accelerate \
3838
datasets \
3939
hf-doc-builder \

docker/diffusers-pytorch-cpu/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ RUN apt install -y bash \
1616
ca-certificates \
1717
libsndfile1-dev \
1818
python3.10 \
19+
python3.10-dev \
1920
python3-pip \
2021
libgl1 \
2122
python3.10-venv && \

docker/diffusers-pytorch-cuda/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ RUN apt install -y bash \
1717
libsndfile1-dev \
1818
libgl1 \
1919
python3.10 \
20+
python3.10-dev \
2021
python3-pip \
2122
python3.10-venv && \
2223
rm -rf /var/lib/apt/lists

docker/diffusers-pytorch-xformers-cuda/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ RUN apt install -y bash \
1717
libsndfile1-dev \
1818
libgl1 \
1919
python3.10 \
20+
python3.10-dev \
2021
python3-pip \
2122
python3.10-venv && \
2223
rm -rf /var/lib/apt/lists

0 commit comments

Comments
 (0)