提交 b789b70c 编写于 作者: 文幕地方's avatar 文幕地方

add system judge

上级 e9555900
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
import argparse import argparse
import os import os
import sys import sys
import platform
import cv2 import cv2
import numpy as np import numpy as np
import paddle import paddle
...@@ -313,6 +314,10 @@ def create_predictor(args, mode, logger): ...@@ -313,6 +314,10 @@ def create_predictor(args, mode, logger):
def get_infer_gpuid(): def get_infer_gpuid():
sysstr = platform.system()
if sysstr == "Windows":
return 0
if not paddle.fluid.core.is_compiled_with_rocm(): if not paddle.fluid.core.is_compiled_with_rocm():
cmd = "env | grep CUDA_VISIBLE_DEVICES" cmd = "env | grep CUDA_VISIBLE_DEVICES"
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册