SegmenTron icon indicating copy to clipboard operation
SegmenTron copied to clipboard

Published 20 hours ago • verified publisher icon LikeLy-Journey

Multi-GPU Training Error

Open xiaodongyang opened this issue 4 years ago • 4 comments

Thanks for the great work! I'm trying to use the multi-gpu training but got an error at line.

RuntimeError: Tensors must be CUDA and dense

xiaodongyang avatar Sep 11 '20 07:09 xiaodongyang

I guess this error relates to pytorch version (1.4 I'm currently using). Could you please let me know what exact version did you use to be able to run the the multi-gpu training?

xiaodongyang avatar Sep 12 '20 07:09 xiaodongyang

i have the same problem when using the multi-gpu training ,i use pytorch1.1

MegumiOuO avatar Sep 13 '20 06:09 MegumiOuO

@Megumi-cs Thanks for the info. Did you figure out a solution?

xiaodongyang avatar Sep 13 '20 08:09 xiaodongyang

~.~ waiting soluting....

jiawenhao2015 avatar Sep 29 '20 09:09 jiawenhao2015