BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard

如何指定使用GPU=[3,4,5,6]训练,不使用GPU=0?

Open renliao opened this issue 1 year ago • 1 comments

亲爱的作者,是配置--local_rank=3吗?我配置了以后发现还是会使用GPU=0.

renliao avatar May 18 '24 03:05 renliao

最简单的方法,在运行命令前加上CUDA_VISIBLE_DEVICES=[3,4,5,6]

Windrain7 avatar Nov 03 '24 12:11 Windrain7