From eacd950689cdd71a1f0dc1a3ea834095eb769ea1 Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Wed, 17 Jul 2024 13:02:55 -0700 Subject: [PATCH] [docs] Fixes pytorch version in README (#3347) --- docker/spark/Dockerfile | 8 ++++---- engines/pytorch/pytorch-engine/README.md | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/docker/spark/Dockerfile b/docker/spark/Dockerfile index dee867ca004..47c1f460a31 100644 --- a/docker/spark/Dockerfile +++ b/docker/spark/Dockerfile @@ -15,10 +15,10 @@ LABEL maintainer="djl-dev@amazon.com" # Install dependencies ARG DJL_VERSION=0.29.0 ARG JNA_VERSION=5.14.0 -ARG JAVACV_VERSION=1.5.9 -ARG JAVACPP_VERSION=1.5.9 -ARG FFMPEG_VERSION=6.0-1.5.9 -ARG TENSORFLOW_CORE_VERSION=0.5.0 +ARG JAVACV_VERSION=1.5.10 +ARG JAVACPP_VERSION=1.5.10 +ARG FFMPEG_VERSION=6.1.1-1.5.10 +ARG TENSORFLOW_CORE_VERSION=1.0.0-rc.1 ARG PROTOBUF_VERSION=3.25.3 ARG PYTORCH_VERSION=2.3.1 diff --git a/engines/pytorch/pytorch-engine/README.md b/engines/pytorch/pytorch-engine/README.md index 6e1495b6bbe..a6fd6f0a920 100644 --- a/engines/pytorch/pytorch-engine/README.md +++ b/engines/pytorch/pytorch-engine/README.md @@ -24,7 +24,7 @@ The javadocs output is built in the `build/doc/javadoc` folder. ## Installation You can pull the PyTorch engine from the central Maven repository by including the following dependency: -- ai.djl.pytorch:pytorch-engine:0.28.0 +- ai.djl.pytorch:pytorch-engine:0.29.0 ```xml @@ -117,7 +117,7 @@ export PYTORCH_FLAVOR=cpu For macOS, you can use the following library: - ai.djl.pytorch:pytorch-jni:2.3.1-0.29.0 -- ai.djl.pytorch:pytorch-native-cpu:2.1.1:osx-x86_64 +- ai.djl.pytorch:pytorch-native-cpu:2.3.1:osx-x86_64 ```xml