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

Reply via email to