tqchen commented on a change in pull request #4826: [CI][DOCKER] Update ci-gpu torch1.4 and onnx1.6 URL: https://github.com/apache/incubator-tvm/pull/4826#discussion_r375626423
########## File path: docker/install/ubuntu_install_onnx.sh ########## @@ -21,7 +21,7 @@ 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 Review comment: nice catch, the binary image is correct, somehow i missed the patch. ``` Type "help", "copyright", "credits" or "license" for more information. >>> import torch i>>> import torchvision >>> torch.__version__ '1.4.0' >>> torchvision.__version__ '0.5.0' >>> ``` ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services