Skip to content

Commit

Permalink
remove pillow simd
Browse files Browse the repository at this point in the history
  • Loading branch information
mvpatel2000 committed Jun 18, 2024
1 parent eb2b59d commit acba011
Showing 1 changed file with 0 additions and 21 deletions.
21 changes: 0 additions & 21 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,6 @@ ARG PYTORCH_VERSION=1.13.1
# version that corresponds to the PyTorch version
ARG TORCHVISION_VERSION=0.14.1

# In the Dockerimage, Pillow-SIMD is installed instead of Pillow. To trick pip into thinking that
# Pillow is also installed (so it won't override it with a future pip install), a Pillow stub is included
# PILLOW_PSEUDOVERSION is the Pillow version that pip thinks is installed
# PILLOW_SIMD_VERSION is the actual version of pillow-simd that is installed.
ARG PILLOW_PSEUDOVERSION=9.3.0
ARG PILLOW_SIMD_VERSION=9.0.0.post1

# Version of the Mellanox Drivers to install (for InfiniBand support)
# Leave blank for no Mellanox Drivers
ARG MOFED_VERSION=5.5-1.0.3.2
Expand Down Expand Up @@ -181,20 +174,6 @@ RUN add-apt-repository ppa:deadsnakes/ppa && \
RUN curl -fsSL https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} - && \
pip${PYTHON_VERSION} install --no-cache-dir --upgrade 'pip<23' 'setuptools<70.0.0'

#####################
# Install pillow-simd
#####################
ARG PILLOW_PSEUDOVERSION
ARG PILLOW_SIMD_VERSION

# pillow_stub tricks pip into thinking that it installed pillow,
# so when pillow_simd is installed, other packages won't later override it
COPY pillow_stub /tmp/pillow_stub

RUN pip${PYTHON_VERSION} install --no-cache-dir --upgrade /tmp/pillow_stub && \
pip${PYTHON_VERSION} install --no-cache-dir --upgrade pillow_simd==${PILLOW_SIMD_VERSION} && \
rm -rf /tmp/pillow_stub

#################
# Install Pytorch
#################
Expand Down

0 comments on commit acba011

Please sign in to comment.