diff --git a/gpu_mnist_example/train_gcu.py b/gpu_mnist_example/train_gcu.py
index 149ec3f..eb52a2a 100644
--- a/gpu_mnist_example/train_gcu.py
+++ b/gpu_mnist_example/train_gcu.py
@@ -45,6 +45,7 @@ from torch.utils.data import DataLoader
 from torchvision.transforms import ToTensor
 import argparse
 import os
+os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH")))
 
 import importlib.util