|
|
|
|
|
|
by agibsonccc
3066 days ago
|
|
|
(Disclaimer: I created and maintain this library which is now apart of the eclipse foundation): http://nd4j.org/ - in built GPU garbage collector and everything. If you want raw cuda primitives (not generally recommendended and hard to do right) - you can take a look at our javacpp based (we also maintain this) cuda bindings:
https://github.com/bytedeco/javacpp-presets/tree/master/cuda Unlike jcuda (which people typically recommend despite not being updated as often) we actually depend on this for the nd4j and deeplearning4j projects. These cuda bindings are meant to be a 1 to 1 mapping to the cuda api as well. Hope this helps! If you want a fairly small and minimalistic look at the underlying c code which uses cuda take a look at:
https://github.com/deeplearning4j/libnd4j All of this is published on maven central for you and runs on linux, windows and even mac. It's also the same api. All you do is switch the backend. |
|
|