fix get gpu id

4fb9b76f · LDOUBLEV · a09bcf5c · 4fb9b76f
显示空白变更内容
内联并排

Showing with 0 addition and 4 deletion

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

未找到文件。
--- 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: