V2AI / Det3D

World's first general purpose 3D object detection codebse.
https://arxiv.org/abs/1908.09492
Apache License 2.0
1.48k stars 299 forks source link

How to specify the gpus used #105

Closed Kins1ley closed 4 years ago

Kins1ley commented 4 years ago

I tried to add os.envrions in the code but it doesn't make sense.

image

Here is the bug

poodarchu commented 4 years ago

GPU is alrealy in use, but it encounters an OOM error, try to reduce your batch size.