experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix packaging for Tensorflow JNI GPU package.

Browse Source 
The equivalent fix for the CPU JNI package was submitted as
aa6702aa1c.

PiperOrigin-RevId: 262023368
This commit is contained in:
James Ring 2019-08-06 16:44:31 -07:00 committed by TensorFlower Gardener
parent 5f30d1a31a
commit d8655c90d9
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/java/maven/run_inside_container.sh
View File

@ -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