[机器学习]如何使用Java调用Libtorch(Pytorch的C++版本)

作为一个Java开发者(虽然我也用Python,但是自从有一次采集数据的时候Python不靠谱,而Java能够更靠谱的将数据采集过来之后,我就怎么想用Python了)自然是想要Python能玩的东西,我Java也能够玩耍。而PyTorch作为机器学习的入门学习包,自然是不容错过。今天阿福就来带大家配置一下能让Java使用LibTorch。

国内一群狗贼,天天抄,基本上没个正行,没有靠谱的教程,我TMD还得靠英文谷歌才谷歌出个教程来。

首先我们先弄个环境变量:

由于我的LibTorch 解压在了D:\sdk\libtorch 下面,因此我的环境变量也就是这么配置的

 

LIBTORCH_HOME=D:\sdk\libtorch

实际Windows下怎么配置如上图。Linux配置的话,我就不多说了,相信用Linux的小伙伴应该很容易配置这个东西。

在配置好LIBTORCH_HOME的变量后,将lib路径添加进环境变量中如下图

这样我们就能通过环境变量来访问libtorch了。

然后我们通过pom 引入相关的Jar包

        
            org.pytorch
            pytorch_java_only
            1.9.0
        

注意,这里的版本号需要跟libtorch的版本号是一样的,不同会出问题

至此就简简单单的就实现了Java调用LibTorch了。妈蛋,想当年傻傻的用百度去搜这玩意,搜出来的都是一堆什么玩意。然后又想通过JNI直接调用。妈蛋现在才发现pytorch官网已经给你做好了JNI的调用。基本上你不用多操作了。真开心。

未经允许不得转载:RffanLAB|Rffan实验室 » [机器学习]如何使用Java调用Libtorch(Pytorch的C++版本)

赞 (113) 打赏

评论 0

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏