diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index 2b893c4..5964908 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -12,7 +12,7 @@ If there are Chinese comments in the code，please add at the beginning：
 
 '''
 import os
-print(os.listdir('/code'))
+os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH")))
 
 from model import Model
 import numpy as np