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 \
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
2727ENV 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 \
Original file line number Diff line number Diff 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 && \
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments