diff --git a/tools/infer/utility.py b/tools/infer/utility.py index af6c54bb1cc8aeadd1687df5fbd70c16c2cc8b4b..bd9e14a65749f4223eeb6cf79a37546909854d17 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: