Skip to content

Commit acdde8c

Browse files
authored
Merge pull request #96 from ThoughtWorksInc/cuda8
Add cuda dependencies for benchmark
2 parents c34f558 + cd01bfa commit acdde8c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmarks/build.sbt

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
enablePlugins(JmhPlugin)
22

3-
libraryDependencies += "org.nd4j" % "nd4j-api" % "0.9.1"
3+
libraryDependencies += "org.nd4j" % "nd4j-api" % "0.8.0"
44

5-
libraryDependencies += "org.nd4j" % "nd4j-native-platform" % "0.9.1"
5+
libraryDependencies += "org.nd4j" % "nd4j-cuda-8.0-platform" % "0.8.0"
6+
7+
libraryDependencies += "org.nd4j" % "nd4j-native-platform" % "0.8.0"
68

79
libraryDependencies += ("org.lwjgl" % "lwjgl" % "3.1.6").jar().classifier {
810
import scala.util.Properties._

0 commit comments

Comments
 (0)