Go to file
shugeo f083b96c74
Shugeo cuda tests (#116)
* Added tests for get_seed/set_seed ops.

* Added missed tests for scatter_sub/mul/div ops.

* Added tests for hardsigmoid and hardsigmoid_bp.

* Added tests for hardtanh and hardtanh_bp ops.

* Added test for histogram op.

* Added tests for identity op.

* Refactored mergemaxindex op. Added tests for log1p,mergemaxindex, mod and mod_bp ops.

* Fixed tests for FloorDiv.

* Added test for rank op.

* Added tests for rationaltanh/rationaltanh_bp ops.

* Added tests for realdiv/realdiv_bp.

* Added tests for rectifiedtanh/_bp ops.

* Added tests for shapes_of op.

* Added tests for shapes_of op.

* Added tests for size op.

* Added tests for softplus/_bp ops.

* Added tests for softsign/_bp ops.

* Added tests for toggle_bits op. Fixed processing of OP_IMPL and so on defititions.

* Added test for truncatediv op.

* Added another test for truncatediv op.

* Added another test for histogram.

* Added tests for unstack_list op.

* Refactored to_int32/uint32/float16/float32/double/int64/uint64 ops and tests.

* Refactored mergemaxindex op helper for cuda platform and tests.

* Fixed cuda kernel for histogram op helper.

* Refactor skipgram to avoid early buffers shift.

* Fixed check up with non_max_suppression op cuda helper. Added cuda kernel implementation for skipgram op helpers.

* Added implementation of skipgram op helper for cuda platform. Working revision

* Fixed mergeMaxIndex kernel and move it to separate source file.
2019-08-15 13:54:47 +03:00
.github Update contributing and issue/PR templates (#7934) 2019-06-22 16:21:27 +10:00
arbiter SameDiff: Listener changes and training api update (#99) 2019-08-10 15:30:31 +10:00
datavec [WIP] Remote inference (#96) 2019-08-14 12:11:09 +03:00
deeplearning4j [WIP] Remote inference (#96) 2019-08-14 12:11:09 +03:00
docs Quick start ND4J (#7916) 2019-07-02 13:43:07 +10:00
gym-java-client Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
jumpy Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
libnd4j Shugeo cuda tests (#116) 2019-08-15 13:54:47 +03:00
nd4j Nd4j refactoring (#112) 2019-08-15 13:50:52 +09:00
nd4s Small number of fixes (#98) 2019-08-05 11:28:07 +10:00
pydatavec Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
pydl4j Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
rl4j RL4J - Added a unit test to help refac QLearningDiscrete.trainStep() (#8065) 2019-08-02 12:50:28 +10:00
scalnet Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
.gitignore Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
CONTRIBUTING.md Update contributing and issue/PR templates (#7934) 2019-06-22 16:21:27 +10:00
Jenkinsfile Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
LICENSE Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
README.md Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
change-cuda-versions.sh Update dependencies to just released JavaCPP and JavaCV 1.5.1 (#8004) 2019-07-14 21:07:33 +03:00
change-scala-versions.sh Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
change-spark-versions.sh Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
perform-release.sh Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
pom.xml Adapt the Java wrappers in ND4J generated with JavaCPP 2019-08-05 11:22:59 +10:00

README.md

Monorepo of Deeplearning4j

Welcome to the new monorepo of Deeplearning4j that contains the source code for all the following projects, in addition to the original repository of Deeplearning4j moved to deeplearning4j:

To build everything, we can use commands like

./change-cuda-versions.sh x.x
./change-scala-versions.sh 2.xx
./change-spark-versions.sh x
mvn clean install -Dmaven.test.skip -Dlibnd4j.cuda=x.x -Dlibnd4j.compute=xx

or

mvn -B -V -U clean install -pl '!jumpy,!pydatavec,!pydl4j' -Dlibnd4j.platform=linux-x86_64 -Dlibnd4j.chip=cuda -Dlibnd4j.cuda=9.2 -Dlibnd4j.compute=<your GPU CC> -Djavacpp.platform=linux-x86_64 -Dmaven.test.skip=true

An example of GPU "CC" or compute capability is 61 for Titan X Pascal.

Want some examples?

We have separate repository with various examples available: https://github.com/deeplearning4j/dl4j-examples

In the examples repo, you'll also find a tutorial series in Zeppelin: https://github.com/deeplearning4j/dl4j-examples/tree/master/tutorials