Merge pull request #5075 from WenmuZhou/shijiang

add system judge
2026-09-24 23:33:08 +08:00 · 2021-12-27 21:13:11 -06:00
parent 3906c9cd06 b789b70cf5
commit d0c85ddc00
1 changed files with 5 additions and 0 deletions
@@ -15,6 +15,7 @@
 import argparse
 import os
 import sys
+import platform
 import cv2
 import numpy as np
 import paddle
@@ -313,6 +314,10 @@ def create_predictor(args, mode, logger):


 def get_infer_gpuid():
+    sysstr = platform.system()
+    if sysstr == "Windows":
+        return 0
+
    if not paddle.fluid.core.is_compiled_with_rocm():
        cmd = "env | grep CUDA_VISIBLE_DEVICES"
    else: