diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py index 5964908..2fd80c8 100644 --- a/gpu_mnist_example/train_gpu.py +++ b/gpu_mnist_example/train_gpu.py @@ -12,6 +12,7 @@ If there are Chinese comments in the code,please add at the beginning: ''' import os +os.system("pip cache purge") os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH"))) from model import Model