Testing TensorFlow Java bindings within Scala
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

23 lines
870 B

lazy val deps = new {
val keras = "0.1.0-SNAPSHOT"
val log = "0.1.1"
val tensorflow = "0.4.0"
}
// change this according to your OS and CPU
val tfClassifier = "linux-x86_64"
lazy val root = project.in(file("."))
.settings(
name := "tensorflow-java-test",
organization := "de.sciss",
licenses := Seq("LGPL v2.1+" -> url("http://www.gnu.org/licenses/lgpl-2.1.txt")),
scalaVersion := "3.1.1",
libraryDependencies ++= Seq(
"de.sciss" %% "log" % deps.log,
"de.sciss" %% "tensorflow-keras" % deps.keras,
"org.tensorflow" % "tensorflow-core-api" % deps.tensorflow,
"org.tensorflow" % "tensorflow-core-api" % deps.tensorflow classifier tfClassifier, // "linux-x86_64-mkl",
"org.tensorflow" % "tensorflow-framework" % deps.tensorflow,
)
)