fix set_device twice bug (#818)

Author: Ash-Lee233

mindocr/models/backbones/yolov8_backbone.py

@@ -1,3 +1,5 @@
+import os
+
 import numpy as np
 
 import mindspore as ms
@@ -119,7 +121,7 @@ def yolov8_backbone(
 
 def test_yolo_backbone():
     ms.set_context(mode=ms.PYNATIVE_MODE)
-    ms.set_device("Ascend", 3)
+    ms.set_device("Ascend", os.environ.get("DEVICE_ID", 0))
     ms.set_seed(0)
 
     network = YOLOv8Backbone()

mindocr/models/heads/yolov8_head.py

@@ -1,4 +1,5 @@
 import math
+import os
 
 import numpy as np
 
@@ -125,7 +126,7 @@ def yolov8_head(nc=5, reg_max=16, stride=None, in_channels=None) -> YOLOv8Head:
 
 def test_yolov8_head():
     ms.set_context(mode=ms.PYNATIVE_MODE)
-    ms.set_device("Ascend", 3)
+    ms.set_device("Ascend", os.environ.get("DEVICE_ID", 0))
     ms.set_seed(0)
 
     network = yolov8_head()