diff --git a/npu_mnist_example/train_npu.py b/npu_mnist_example/train_npu.py
index 194f230..5445dba 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 {}".format(os.getenv("OPENI_SDK_PATH")))
 import argparse
 from config import mnist_cfg as cfg
 from dataset import create_dataset