mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix get gpu id

pull/4752/head
LDOUBLEV 2021-11-25 09:58:50 +08:00
parent a09bcf5c21
commit 4fb9b76fff
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/infer/utility.py
View File

@ -311,10 +311,6 @@ def create_predictor(args, mode, logger):
def get_infer_gpuid(): def get_infer_gpuid():
#cmd = "nvidia-smi"
#res = os.popen(cmd).readlines()
#if len(res) == 0:
# return None
cmd = "env | grep CUDA_VISIBLE_DEVICES" cmd = "env | grep CUDA_VISIBLE_DEVICES"
env_cuda = os.popen(cmd).readlines() env_cuda = os.popen(cmd).readlines()
if len(env_cuda) == 0: if len(env_cuda) == 0: