Browse Source

更新 'gpu_mnist_example/train_gcu.py'

liuzx-patch-1
liuzxtest02 2 years ago
parent
commit
9a5a21db7a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      gpu_mnist_example/train_gcu.py

+ 1
- 0
gpu_mnist_example/train_gcu.py View File

@@ -31,6 +31,7 @@ If there are Chinese comments in the code,please add at the beginning:

import os
print("begin:")
os.system("pip uninstall openi-test")
os.system("pip install {}".format(os.getenv("OPENI_SDK_PATH")))
import torch
from model import Model


Loading…
Cancel
Save