From d8655c90d9b8f346bb86b14a6a8cb9429515e5c3 Mon Sep 17 00:00:00 2001 From: James Ring Date: Tue, 6 Aug 2019 16:44:31 -0700 Subject: [PATCH] Fix packaging for Tensorflow JNI GPU package. The equivalent fix for the CPU JNI package was submitted as https://github.com/tensorflow/tensorflow/commit/aa6702aa1cb9a1f2a72446cd282dc607619539f8. PiperOrigin-RevId: 262023368 --- tensorflow/java/maven/run_inside_container.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/java/maven/run_inside_container.sh b/tensorflow/java/maven/run_inside_container.sh index 2598eb0e457..3899ebbf1f4 100644 --- a/tensorflow/java/maven/run_inside_container.sh +++ b/tensorflow/java/maven/run_inside_container.sh @@ -120,6 +120,11 @@ download_libtensorflow_jni_gpu() { curl -L "${RELEASE_URL_PREFIX}/libtensorflow_jni-gpu-linux-x86_64-${TF_VERSION}.tar.gz" | tar -xvz -C linux-x86_64 curl -L "${RELEASE_URL_PREFIX}/libtensorflow_jni-gpu-windows-x86_64-${TF_VERSION}.zip" -o /tmp/windows.zip + FRAMEWORK_SO="$(readlink -f linux-x86_64/libtensorflow_framework.so)" + rm linux-x86_64/libtensorflow_framework.so + rm "linux-x86_64/libtensorflow_framework.so.${MAJOR_VERSION}" + mv "${FRAMEWORK_SO}" "linux-x86_64/libtensorflow_framework.so.${MAJOR_VERSION}" + unzip /tmp/windows.zip -d windows-x86_64 rm -f /tmp/windows.zip