diff --git a/benchmarks/build.sbt b/benchmarks/build.sbt
index 90f2b0b8..208973f6 100644
--- a/benchmarks/build.sbt
+++ b/benchmarks/build.sbt
@@ -1,6 +1,6 @@
 enablePlugins(JmhPlugin)
 
-libraryDependencies += "org.nd4j" % "nd4j-api" % "0.8.0"
+libraryDependencies += "org.nd4j" % "nd4j-api" % "0.9.1"
 
 val nd4jRuntime = settingKey[String]("\"cuda-8.0\" to run benchmark on GPU, \"native\" to run benchmark on CPU.")
 
@@ -8,7 +8,7 @@ nd4jRuntime in Global := "native"
 
 libraryDependencies += "ch.qos.logback" % "logback-classic" % "1.2.3"
 
-libraryDependencies += "org.nd4j" % s"nd4j-${nd4jRuntime.value}-platform" % "0.8.0"
+libraryDependencies += "org.nd4j" % s"nd4j-${nd4jRuntime.value}-platform" % "0.9.1"
 
 libraryDependencies += ("org.lwjgl" % "lwjgl" % "3.2.2").jar().classifier {
   import scala.util.Properties._