diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 57442a99abc2..b0627688eca8 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -87,7 +87,7 @@ RUN bash /install/ubuntu_install_caffe2.sh COPY install/ubuntu_install_dgl.sh /install/ubuntu_install_dgl.sh RUN bash /install/ubuntu_install_dgl.sh -RUN pip3 install Pillow +RUN pip3 install "Pillow<7" COPY install/ubuntu_install_vulkan.sh /install/ubuntu_install_vulkan.sh RUN bash /install/ubuntu_install_vulkan.sh diff --git a/docker/Dockerfile.demo_gpu b/docker/Dockerfile.demo_gpu index 0591050c5270..8ab174aac8e7 100644 --- a/docker/Dockerfile.demo_gpu +++ b/docker/Dockerfile.demo_gpu @@ -21,7 +21,7 @@ FROM tvmai/ci-gpu:v0.54 # Jupyter notebook. -RUN pip3 install matplotlib Image Pillow jupyter[notebook] +RUN pip3 install matplotlib Image "Pillow<7" jupyter[notebook] # Build TVM COPY install/install_tvm_gpu.sh /install/install_tvm_gpu.sh diff --git a/docker/Dockerfile.demo_opencl b/docker/Dockerfile.demo_opencl index bf27eef862f5..fb2d8999dc54 100644 --- a/docker/Dockerfile.demo_opencl +++ b/docker/Dockerfile.demo_opencl @@ -45,7 +45,7 @@ RUN apt-get install -y python3-dev python3-pip RUN pip3 install setuptools numpy pytest cython decorator scipy tornado psutil xgboost RUN echo "Installing Jupyter notebook" -RUN pip3 install matplotlib Image Pillow jupyter[notebook] +RUN pip3 install matplotlib Image "Pillow<7" jupyter[notebook] RUN echo "Installing OpenCL libraries" RUN apt-get install -y libviennacl-dev mesa-opencl-icd ocl-icd-opencl-dev clinfo diff --git a/docker/install/ubuntu_install_python_package.sh b/docker/install/ubuntu_install_python_package.sh index 2eaf00e8fdd0..32fa0b2251a0 100755 --- a/docker/install/ubuntu_install_python_package.sh +++ b/docker/install/ubuntu_install_python_package.sh @@ -21,4 +21,4 @@ set -u set -o pipefail # install libraries for python package on ubuntu -pip3 install pylint==1.9.4 six numpy pytest cython decorator scipy tornado typed_ast pytest mypy orderedset antlr4-python3-runtime attrs requests Pillow packaging +pip3 install pylint==1.9.4 six numpy pytest cython decorator scipy tornado typed_ast pytest mypy orderedset antlr4-python3-runtime attrs requests "Pillow<7" packaging diff --git a/docs/README.txt b/docs/README.txt index fffdaa233ef8..18d97a3949f5 100644 --- a/docs/README.txt +++ b/docs/README.txt @@ -3,7 +3,7 @@ TVM Documentations This folder contains the source of TVM documents - A hosted version of doc is at http://docs.tvm.ai -- pip install sphinx>=1.5.5 sphinx-gallery sphinx_rtd_theme matplotlib Image recommonmark Pillow +- pip install sphinx>=1.5.5 sphinx-gallery sphinx_rtd_theme matplotlib Image recommonmark "Pillow<7" - Build tvm first in the root folder. - To build locally, you need to enable USE_CUDA, USE_OPENCL, LLVM_CONFIG in config.mk and then type "make html" in this folder. diff --git a/python/setup.py b/python/setup.py index bc53060f95cf..4e4f7034f750 100644 --- a/python/setup.py +++ b/python/setup.py @@ -160,7 +160,7 @@ def get_package_data_files(): 'attrs', 'psutil', ], - extras_require={'test': ['pillow', + extras_require={'test': ['pillow<7', 'matplotlib'], 'extra_feature': ['tornado', 'psutil', diff --git a/vta/tutorials/autotvm/tune_relay_vta.py b/vta/tutorials/autotvm/tune_relay_vta.py index a9ab6d70ae48..eab23eee0b0f 100644 --- a/vta/tutorials/autotvm/tune_relay_vta.py +++ b/vta/tutorials/autotvm/tune_relay_vta.py @@ -40,7 +40,7 @@ # # .. code-block:: bash # -# pip3 install --user psutil xgboost tornado mxnet requests pillow +# pip3 install --user psutil xgboost tornado mxnet requests "Pillow<7" # # To make TVM run faster during tuning, it is recommended to use cython # as FFI of TVM. In the root directory of TVM, execute diff --git a/vta/tutorials/frontend/deploy_vision_on_vta.py b/vta/tutorials/frontend/deploy_vision_on_vta.py index a3169865e931..47773bda5ec7 100644 --- a/vta/tutorials/frontend/deploy_vision_on_vta.py +++ b/vta/tutorials/frontend/deploy_vision_on_vta.py @@ -34,7 +34,7 @@ # # .. code-block:: bash # -# pip3 install --user mxnet requests pillow +# pip3 install --user mxnet requests "Pillow<7" # # Now return to the python code. Import packages.