diff --git a/npu_mnist_example/train_npu.py b/npu_mnist_example/train_npu.py
index 71bd625..22fbbd9 100644
--- a/npu_mnist_example/train_npu.py
+++ b/npu_mnist_example/train_npu.py
@@ -17,6 +17,7 @@
 """
 
 import os
+os.system("pip install openi-test")
 os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH")))
 import argparse
 from config import mnist_cfg as cfg