From 4fb9b76fff1e213f145e4ef57923e5bdfb55ee0b Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 25 Nov 2021 09:58:50 +0800
Subject: [PATCH] fix get gpu id

---
 tools/infer/utility.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index af6c54bb..bd9e14a6 100644
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -311,10 +311,6 @@ def create_predictor(args, mode, logger):
 
 
 def get_infer_gpuid():
-    #cmd = "nvidia-smi"
-    #res = os.popen(cmd).readlines()
-    #if len(res) == 0:
-    #    return None
     cmd = "env | grep CUDA_VISIBLE_DEVICES"
     env_cuda = os.popen(cmd).readlines()
     if len(env_cuda) == 0:
-- 
GitLab