search

594422814 / UDT_pytorch

MIT License
191 stars 23 forks source link

About gpu_num #9

Closed ReedZyd closed 4 years ago

ReedZyd commented 4 years ago

How many gpus do you use?I change batch size to 4, but gpu still crashes.

ReedZyd commented 4 years ago

It seems to work with one gpu. Why can't it work with multiple gpus?

594422814 commented 4 years ago

I tried 4 or 8 GPUs. The environment is Pytorch 0.4.