前提
1.确保你本机具有gpu硬件支持;
2.已经安装gpu软件环境;
3.java工程除无法正常加载gpu外可正常运行。
4.此博客是在Ubuntu16.04上搭建java gpu运行环境,若在Windows上搭建,则在满足前三条的前提下,将下面提到的jar包解压后的.dll文件进行手动加载即可。
maven依赖
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>1.10.0</version>
</dependency>
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow</artifactId>
<version>1.10.0</version>
</dependency>
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow_jni_gpu</artifactId>
<version>1.10.0</version>
</dependency>