CUDA_VISIB..GPU:RTX3060 * 2TF:2.4.1CUDA:11.1CUDNN:8.0.5问题是不论设置CUDA_VISIBLE_DEVICES为1还是0,都会把任务建在GPU0上。CUDA_VISIBL
问题描述 / Problem Description os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus 指定要使用的显卡不起作用,还是使用所有的显卡 复现问题的步骤 / Steps to Reproduce 在不同位置打印torch.cuda.device_count(),发现os.environ["CUDA_VISIBLE_DEVICES"] = args.gpus不起