search

notAI-tech / NudeNet

Lightweight nudity detection
https://nudenet.notai.tech/
GNU Affero General Public License v3.0
1.81k stars 351 forks source link

onnxruntime gpu support #71

Closed shady145 closed 3 years ago

shady145 commented 3 years ago

Will we be able to use the GPU now that nudenet uses onnxruntime ?

bedapudi6788 commented 3 years ago

Yes. you can just uninstall onnxruntime and install onnxruntime-gpu if you want to use gpu.

shady145 commented 3 years ago

When using onnxruntime-gpu im getting ModuleNotFoundError: No module named 'onnxruntime'. Can't find anything about it online. Any clue ?

bedapudi6788 commented 3 years ago

Can you reproduce the issue in a colab notebook and post it here?

On Wed, 27 Jan 2021, 22:26 shady145, notifications@github.com wrote:

When using onnxruntime-gpu im getting ModuleNotFoundError: No module named 'onnxruntime'. Can't find anything about it online. Any clue ?

— You are receiving this because you modified the open/close state. Reply to this email directly, view it on GitHub https://github.com/notAI-tech/NudeNet/issues/71#issuecomment-768423782, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADZJQHSJQJRHFSA2DUQGAR3S4BAS7ANCNFSM4WVD42ZA .