From 1e287ef24faeaecda51f97ae83bd471bb5492161 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Thu, 6 Feb 2020 11:58:11 -0800
Subject: [PATCH] [CI][DOCKER] Update ci-gpu to v0.60 (#4827)

---
 Jenkinsfile                                | 2 +-
 tests/python/frontend/onnx/test_forward.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index 7a2afb3fdf87..92d9e535963d 100644
--- 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 fc05e7a011a9..f9e8aac4e389 100644
--- 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