不要使用CUDA_VISIBLE_DEVICES来选择要是用的GPU,要直接在Trainer里面的gpus来选择要用的GPU。
Change from:
CUDA_VISIBLE_DEVICES=0,1
To:
trainer = Trainer(...
gpus='0, 1',
...)
不要使用CUDA_VISIBLE_DEVICES来选择要是用的GPU,要直接在Trainer里面的gpus来选择要用的GPU。
Change from:
CUDA_VISIBLE_DEVICES=0,1
To:
trainer = Trainer(...
gpus='0, 1',
...)