在pytorch代码中,os.environ[‘CUDA_VISIBLE_DEVICES’]=num无效,可以在linux命令行运行py文件时输入CUDA_VISIBLE_DEVICES=num1,num2 python 文件名.py
可以通过torch.cuda.device_count()查看visible的gpu的数量
在pytorch代码中,os.environ[‘CUDA_VISIBLE_DEVICES’]=num无效,可以在linux命令行运行py文件时输入CUDA_VISIBLE_DEVICES=num1,num2 python 文件名.py
可以通过torch.cuda.device_count()查看visible的gpu的数量