java对libtensorflow_jni_gpu依赖不能自动加载的解决

本文档介绍了在Ubuntu 16.04上遇到Java无法自动加载libtensorflow_jni_gpu的问题及解决方案。在确认GPU硬件及软件环境正常后,通过手动解压maven依赖的jar包,加载libtensorflow_framework.so和libtensorflow_jni.so,并在IDEA的运行配置中设置`-Djava.library.path`,确保JVM能识别到这些库。同时,在项目启动时使用TensorFlow的loadLibrary方法手动加载GPU库。
摘要由CSDN通过智能技术生成

前提

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>

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值