diff --git a/Jenkinsfile b/Jenkinsfile
index 1cd8575fa3fd..6694b0f136a0 100755
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -46,7 +46,7 @@ import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
 
 // NOTE: these lines are scanned by docker/dev_common.sh. Please update the regex as needed. -->
 ci_lint = "tlcpack/ci-lint:v0.67"
-ci_gpu = "tlcpack/ci-gpu:v0.80"
+ci_gpu = "tlcpack/ci-gpu:v0.81"
 ci_cpu = "tlcpack/ci-cpu:v0.80"
 ci_wasm = "tlcpack/ci-wasm:v0.71"
 ci_i386 = "tlcpack/ci-i386:v0.74"
diff --git a/gallery/how_to/work_with_relay/build_gcn.py b/gallery/how_to/work_with_relay/build_gcn.py
index 4352088026ef..d76baec1eec1 100644
--- a/gallery/how_to/work_with_relay/build_gcn.py
+++ b/gallery/how_to/work_with_relay/build_gcn.py
@@ -119,12 +119,6 @@ def evaluate(data, logits):
     dimension of model output (Number of classes)
 """
 dataset = "cora"
-
-# Temporary disable running load_dataset(dataset) until the CI issue is resolved
-import sys
-
-sys.exit()
-
 g, data = load_dataset(dataset)
 
 num_layers = 1