can I use the library with GPU?

josephsdavid / N2D

Library implementation of https://arxiv.org/abs/1908.05968v5

https://n2d.readthedocs.io/en/latest/

MIT License

23 stars 6 forks source link

Closed dvaler01 closed 4 years ago

dvaler01 commented 4 years ago

Hello David,

I installed the n2d library and I run my first demo but it seems to working on CPU. Can I use GPU or the library for now is implemented only for CPU?

Dimitris.

josephsdavid commented 4 years ago

Yes sir! If you have tensorflow with CUDA support installed, it should run on GPU (it does on my computer). You can test if you’re able to use GPU using one of these functions: https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available

Hopefully this gets you fixed up!

josephsdavid commented 4 years ago

I’ll update the installation instructions I think

dvaler01 commented 4 years ago

Hello David,

Everything is fine. The library is working perfectly. Now I am looking the implementation because I am going to customize some functions.

Thank you for your help. :)

josephsdavid commented 4 years ago

Great! Glad everything is together :) let me know if anything can be changed/improved for easier use!