File tree Expand file tree Collapse file tree 4 files changed +10
-7
lines changed
diffusers-pytorch-compile-cuda
diffusers-pytorch-xformers-cuda Expand file tree Collapse file tree 4 files changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -16,24 +16,24 @@ RUN apt install -y bash \
16
16
ca-certificates \
17
17
libsndfile1-dev \
18
18
libgl1 \
19
- python3.9 \
20
- python3.9 -dev \
19
+ python3.10 \
20
+ python3.10 -dev \
21
21
python3-pip \
22
- python3.9 -venv && \
22
+ python3.10 -venv && \
23
23
rm -rf /var/lib/apt/lists
24
24
25
25
# make sure to use venv
26
- RUN python3.9 -m venv /opt/venv
26
+ RUN python3.10 -m venv /opt/venv
27
27
ENV PATH="/opt/venv/bin:$PATH"
28
28
29
29
# 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 \
32
32
torch \
33
33
torchvision \
34
34
torchaudio \
35
35
invisible_watermark && \
36
- python3.9 -m pip install --no-cache-dir \
36
+ python3.10 -m pip install --no-cache-dir \
37
37
accelerate \
38
38
datasets \
39
39
hf-doc-builder \
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ RUN apt install -y bash \
16
16
ca-certificates \
17
17
libsndfile1-dev \
18
18
python3.10 \
19
+ python3.10-dev \
19
20
python3-pip \
20
21
libgl1 \
21
22
python3.10-venv && \
Original file line number Diff line number Diff line change @@ -17,6 +17,7 @@ RUN apt install -y bash \
17
17
libsndfile1-dev \
18
18
libgl1 \
19
19
python3.10 \
20
+ python3.10-dev \
20
21
python3-pip \
21
22
python3.10-venv && \
22
23
rm -rf /var/lib/apt/lists
Original file line number Diff line number Diff line change @@ -17,6 +17,7 @@ RUN apt install -y bash \
17
17
libsndfile1-dev \
18
18
libgl1 \
19
19
python3.10 \
20
+ python3.10-dev \
20
21
python3-pip \
21
22
python3.10-venv && \
22
23
rm -rf /var/lib/apt/lists
You can’t perform that action at this time.
0 commit comments