Skip to content

Commit

Permalink
Docker: use no cache dir for xformers and pillow-simd
Browse files Browse the repository at this point in the history
  • Loading branch information
Trojaner authored May 5, 2023
1 parent cccd417 commit 308d1d5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -22,15 +22,15 @@ RUN python3 -m pip install wheel
RUN python3 -m pip install --no-cache-dir --use-pep517 -U torch>=2.1.0 ninja --extra-index-url https://download.pytorch.org/whl/nightly/cu121

# Install xformers nightly
RUN python3 -m pip pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
RUN python3 -m pip install --no-cache-dir -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers

# Install requirements
COPY requirements.txt setup.py ./
RUN python3 -m pip install --no-cache-dir --use-pep517 -U -r requirements.txt

# Replace pillow with pillow-simd
RUN python3 -m pip uninstall -y pillow && \
CC="cc -mavx2" python3 -m pip install -U --force-reinstall pillow-simd
CC="cc -mavx2" python3 -m pip install --no-cache-dir -U --force-reinstall pillow-simd

# Fix missing libnvinfer7
USER root
Expand Down

0 comments on commit 308d1d5

Please sign in to comment.