From 65d5abcc0c8a62194b88da2f7044540663e23250 Mon Sep 17 00:00:00 2001
From: liuzxtest02 <134442@163.com>
Date: Tue, 14 Nov 2023 15:07:03 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fmnist=5Fexample/tra?=
 =?UTF-8?q?in=5Fgcu.py'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gpu_mnist_example/train_gcu.py | 1 +
 1 file changed, 1 insertion(+)

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