不要使用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',
...)
本文标题：pytorch_lightning 多块GPU训练速度很慢
本文链接：https://www.haomeiwen.com/subject/lbfkfltx.html
网友评论