search

robofeiathome / detector_2d

MIT License
0 stars 1 forks source link

Colocar o processamento para a GPU #1

Closed cauansousa closed 1 year ago

cauansousa commented 1 year ago

O predict da yolo está sendo feita na cpu (por default), e isso faz com que seja muito lento, ainda mais na jetson. Então é preciso realocar o processamento para a GPU

cauansousa commented 1 year ago

There is an specific version to pytorch run in jetson ('1.14.0a0+44dac51c.nv23.01'). That isn't compatible with the current torchvision version. So im trying to find the one.