美文网首页
Tensorflow 2.x Java api的maven包怎么

Tensorflow 2.x Java api的maven包怎么

作者: DS事务所 | 来源:发表于2020-10-09 14:35 被阅读0次

截至发文,tf最新版为2.3.1,以下以2.3.1举例
按照tf 1.15在pom中引入的写法,2.3.1应该是

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow</artifactId>
  <version>2.3.1</version>
</dependency>

但是,遍搜maven,发现tensorflow版本只到1.15.0最后更新时间为 Oct, 2019


难道要自行build install了么

好吧,老办法,放issue

tensorflow issue #42670 [1]可以发现,tensorflow 2.x后,tensorflow java api另起了一个项目叫做 tensorflow/java [2]

这下破案了

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-api</artifactId>
  <version>0.2.0</version>
</dependency>
<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-api</artifactId>
  <version>0.2.0</version>
  <classifier>linux-x86_64${javacpp.platform.extension}</classifier>
</dependency>
注意windows和linux不同

参考
[1]: tensorflow issue #42670
[2]: tensorflow/java

相关文章