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

diff --git a/Jenkinsfile b/Jenkinsfile
index 6b92f96..e65f73e 100644 (file)
--- 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
index fc05e7a..f9e8aac 100644 (file)
--- 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 @@ import tvm
 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