diff --git a/gpu_mnist_example/train_gcu.py b/gpu_mnist_example/train_gcu.py
index 81d70a0..1b4322c 100644
--- a/gpu_mnist_example/train_gcu.py
+++ b/gpu_mnist_example/train_gcu.py
@@ -31,6 +31,7 @@ If there are Chinese comments in the code，please add at the beginning：
 
 import os
 print("begin:")
+os.system("pip uninstall openi-test")
 os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH")))
 import torch
 from model import Model