Skip to content

GitLab

Unverified Commit d0c85ddc authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge pull request #5075 from WenmuZhou/shijiang 

add system judge
parents 3906c9cd b789b70c
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
tools/infer/utility.py
View file @ d0c85ddc
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment