训练核心显示修改-2

app/controller/AlgorithmController.py

@@ -21,7 +21,7 @@ from app.utils.websocket_tool import manager
 import sys
 from pathlib import Path
 
-# from pynvml import *
+from pynvml import *
 # FILE = Path(__file__).resolve()
 # ROOT = FILE.parents[0]  # YOLOv5 root directory
 # if str(ROOT) not in sys.path: