BasicSR
BasicSR copied to clipboard
如何指定使用GPU=[3,4,5,6]训练,不使用GPU=0?
亲爱的作者,是配置--local_rank=3吗?我配置了以后发现还是会使用GPU=0.
最简单的方法,在运行命令前加上CUDA_VISIBLE_DEVICES=[3,4,5,6]