fix get gpu id

4fb9b76f · LDOUBLEV · a09bcf5c · 4fb9b76f
Commit 4fb9b76f authored Nov 25, 2021 by LDOUBLEV
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 4 deletions

tools/infer/utility.py tools/infer/utility.py +0 -4

No files found.
--- 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: