[Docker] update onnx to 1.6 and torch to 1.4

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<7"
+RUN pip3 install Pillow
 
 COPY install/ubuntu_install_vulkan.sh /install/ubuntu_install_vulkan.sh
 RUN bash /install/ubuntu_install_vulkan.sh

docker/Dockerfile.demo_gpu

@@ -21,7 +21,7 @@
 FROM tvmai/ci-gpu:v0.55
 
 # Jupyter notebook.
-RUN pip3 install matplotlib Image "Pillow<7" jupyter[notebook]
+RUN pip3 install matplotlib Image Pillow jupyter[notebook]
 
 # Build TVM
 COPY install/install_tvm_gpu.sh /install/install_tvm_gpu.sh

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<7" jupyter[notebook]
+RUN pip3 install matplotlib Image Pillow jupyter[notebook]
 
 RUN echo "Installing OpenCL libraries"
 RUN apt-get install -y libviennacl-dev mesa-opencl-icd ocl-icd-opencl-dev clinfo

docker/install/ubuntu_install_onnx.sh

@@ -21,11 +21,11 @@ set -u
 set -o pipefail
 
 # fix to certain version for now
-pip3 install onnx==1.5.0
+pip3 install onnx==1.6.0
 pip3 install onnxruntime==1.0.0
 
 # torch depends on a number of other packages, but unhelpfully, does
 # not expose that in the wheel!!!
 pip3 install future
 
-pip3 install torch==1.2.0 torchvision==0.4.0
+pip3 install torch==1.4.0 torchvision==0.5.0

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<7" 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 packaging

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<7"
+- pip install sphinx>=1.5.5 sphinx-gallery sphinx_rtd_theme matplotlib Image recommonmark Pillow
 - 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.
 

python/setup.py

@@ -160,7 +160,7 @@ def get_package_data_files():
         'attrs',
         'psutil',
         ],
-      extras_require={'test': ['pillow<7',
+      extras_require={'test': ['pillow',
                                'matplotlib'],
                       'extra_feature': ['tornado',
                                         'psutil',

tests/python/frontend/onnx/test_forward.py

@@ -16,6 +16,8 @@
 # under the License.
 import numpy as np
 import math
+import onnx
+from onnx import helper, TensorProto, mapping
 import torch
 import torchvision
 import topi
@@ -24,8 +26,6 @@
 from tvm import relay
 from tvm.contrib import graph_runtime
 from tvm.relay.testing.config import ctx_list
-import onnx
-from onnx import helper, TensorProto, mapping
 import scipy
 
 

vta/tutorials/autotvm/tune_relay_vta.py

@@ -40,7 +40,7 @@
 #
 # .. code-block:: bash
 #
-#   pip3 install --user psutil xgboost tornado mxnet requests "Pillow<7"
+#   pip3 install --user psutil xgboost tornado mxnet requests pillow
 #
 # To make TVM run faster during tuning, it is recommended to use cython
 # as FFI of TVM. In the root directory of TVM, execute

vta/tutorials/frontend/deploy_vision_on_vta.py

@@ -34,7 +34,7 @@
 #
 # .. code-block:: bash
 #
-#   pip3 install --user mxnet requests "Pillow<7"
+#   pip3 install --user mxnet requests pillow
 #
 # Now return to the python code. Import packages.