diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index 2fd6118..bb2f938 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -13,7 +13,6 @@ If there are Chinese comments in the code，please add at the beginning：
 '''
 
 import os
-os.system("pip install -U openi_test")
 from model import Model
 import numpy as np
 import torch