diff --git a/main.py b/main.py
index a4989abf58e12a13cad8568c9a6859427dfd0a56..f13b0a55e863480465f63555142a2dfd502d78bf 100644
--- a/main.py
+++ b/main.py
@@ -15,7 +15,7 @@ from dataset import GetDatasetGenerator
 from loss import SoftmaxCrossEntropyLoss
 
 context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False,
-                    device_target='Ascend', device_id=8)
+                    device_target='Ascend', device_id=7)
 net = UNetMedical(n_channels=3, n_classes=6)
 
 train_dataset_generator = GetDatasetGenerator('./datasets', 'train')