[CI][DOCKER] Update ci-gpu to v0.60 (#4827)

apache · Feb 6, 2020 · b88de43 · b88de43
1 parent 62543d4
commit b88de43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -45,7 +45,7 @@
 //
 
 ci_lint = "tvmai/ci-lint:v0.51"
-ci_gpu = "tvmai/ci-gpu:v0.56"
+ci_gpu = "tvmai/ci-gpu:v0.60"
 ci_cpu = "tvmai/ci-cpu:v0.55"
 ci_i386 = "tvmai/ci-i386:v0.52"
 

diff --git a/tests/python/frontend/onnx/test_forward.py b/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