Windows 10 DJL 安装 部署 Java机器学习 Pytorch

Windows 10 DJL 安装 部署 Java机器学习 Pytorch

RT,挺简单一个东西折腾了一天,下面就说下要点。这里默认你安装好了IDEA和Java 1.8,DJL目前最新版是0.15,官网推荐使用java 1.8

  1. 安装anaconda,直接最新版3.9就可以

link https://www.anaconda.com/

  1. 安装完后记得设置环境变量(不懂的搜索ananconda环境变量,有很多教程,成功的标准是再CMD命令窗口输入conda 不报错)

  2. 安装深度学习库(我这里使用的是pytorch)

  3. 到这里去选好要下载的包然后复制下面的命令

  4. https://pytorch.org/get-started/locally/

Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第1张图片
这里虽然我们使用java编写机器学习,但是不用选java/c++因为DJL会自动调用python版本的pytorch的c底层。

  1. 如果用到显卡,就去NVIDA官网CUDA下载安装,有很多教程这里也不讲了
  2. Win+R键盘打开CMD,把刚才的命令粘贴进去,回车就能自动下载pytorch了**
conda install pytorch torchvision torchaudio cpuonly -c pytorch
  1. 最后,安装DJL。

在这之前必须认识两个东西:Maven或者Gradle,否则根本不知道咋安装的DJL!
这两个东西简而言之就是用来构建项目和自动下载并管理你要用的包的东西,以后大公司项目会用到,可以了解下。

  1. 使用Maven或者Gradle都有一个文本用来标记你要下载的包。使用这两个东东只需要修改Maven的pom.xml或者gradle的build.gradle,当然还有很多细节,这里就不讲了。

这里以Gradle+IDEA为例,左边的文件结构就是由Gradle生成的。右边的build.gradle文件就是决定你下载的包,其中compile的第一个包就是我们要安装或者说引入的DJL包。
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第2张图片

建议使用IDEA来创建项目,这样就不用自己手动去下载Gradle或者Maven,
下面是具体步骤,不想知道为什么就照着做行了。
1.创建新IDEA项目(一定要再这个界面,如果你需要配置Gradle或者Maven 的话点击右下角的configue进行配置)
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第3张图片
2.不想配置的话直接点击new project,这里以gradle为例子,选好后点击下一步
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第4张图片
3.设置路径和项目名字,下一步,finish完成创建
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第5张图片
4.这里Idea 会自动帮你打开build.gradle(如果你使用maven就会自动打开pom.xel)。然后把配置代码粘进去

plugins {
    id 'java'
}
repositories {
    mavenCentral()
}

dependencies {
    //testCompile group: 'junit', name: 'junit', version: '4.12'
    compile("ai.djl.pytorch:pytorch-engine:0.15.0")
    //导入slf4j依赖包,导入下面的两个包,其中logback-classic会导入slf4j包到项目的ExternalLibraries。 logback-core知识可以百度。
    compile('ch.qos.logback:logback-core:1.2.3')
    compile('ch.qos.logback:logback-classic:1.2.3')
    //compile('org.slf4j:slf4j-simple:1.7.25')

}

5.点击右上角的蓝色回环箭头来重新构建项目
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第6张图片
6.如果不出意外的话,重新构建完成项目DJL就引入成功了。
然后你就可以在左侧创建自己的项目文件,愉快地调用NDManager进行矩阵运算了。
Windows 10 DJL 安装 部署 Java机器学习 Pytorch_第7张图片

7.最后说一下可能出现的问题。
1.你使用的idea版本不同可能gradle版本也打不同,那么刚才你复制粘贴的那段代码的语法可能会已经被弃用,这时候你就要搜索一下新版的gradle语法是怎么样的,我这个idea是2020.2的,虽然由提示但是还可以用。
2.如果出现了no engine found这个错误,你就要检查下你装的pytorch版本是否和刚才引入的相对应。多了就不说了,搞机器学习一定要有自己解决问题的能力,不行就being一下。

你可能感兴趣的:(java,机器学习,windows)