fix get gpu id

2021-11-25 09:58:50 +08:00 · 2021-11-25 09:58:50 +08:00 · 4fb9b76fff
parent a09bcf5c21
commit 4fb9b76fff
1 changed files with 0 additions and 4 deletions
--- 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: