如何在弹跳库中设置AI

Question

我想在我的项目中使用深度学习java。但是我无法在我的春季启动kotlin Gradle项目中设置AI DJL。我使用Buildgradle.kts，但不知道如何在buildgradle.kts.

中添加ai.djl

Answer 1

DJL提供了springboot启动程序包，您可以在build.gradle.kts中添加以下部分：

dependencies {
    implementation("org.springframework.boot:spring-boot-starter-web")
    implementation("ai.djl.spring:djl-spring-boot-starter-pytorch-auto:0.11")
    implementation("org.springframework.boot:spring-boot-starter-actuator")
    testImplementation("org.springframework.boot:spring-boot-starter-test") {
        exclude(group = "org.junit.vintage", module = "junit-vintage-engine")
    }
    // See: https://github.com/awslabs/djl/blob/master/mxnet/mxnet-engine/README.md for MXNet library selection
}

您可以从github：https://github.com/deepjavalibrary/djl-spring-boot-starter-demo上找到DJL springboot初学者演示项目。