问题描述 / Problem Description os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus 指定要使用的显卡不起作用,还是使用所有的显卡 复现问题的步骤 / Steps to Reproduce 在不同位置打印torch.cuda.device_count(),发现os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus不起