更新 'gpu_mnist_example/train_gcu.py'

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