Skip to content

Commit 754cb96

Browse files
committed
add .so back in docker
1 parent 77e34bf commit 754cb96

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

Dockerfile

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,11 @@ ENV PATH $SPARK_HOME/bin:$PATH
4141
ENV PYTHONPATH /opt/spark/python/lib/py4j-0.10.7-src.zip:/opt/spark/python/lib/pyspark.zip:$PYTHONPATH
4242
ENV PYSPARK_PYTHON python
4343

44+
# Workaround for https://github.com/tensorflow/tensorflow/issues/30635.
45+
RUN wget -q https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-1.15.0-rc3.tar.gz && \
46+
tar xf libtensorflow_jni-cpu-linux-x86_64-1.15.0-rc3.tar.gz && \
47+
cp libtensorflow_framework.so.1 /usr/lib
48+
4449
# The tensorframes dir will be mounted here.
4550
VOLUME /mnt/tensorframes
4651
WORKDIR /mnt/tensorframes

project/Build.scala

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,7 @@ object Shading extends Build {
6363
"com.typesafe.scala-logging" %% "scala-logging-api" % "2.1.2",
6464
"com.typesafe.scala-logging" %% "scala-logging-slf4j" % "2.1.2",
6565
// TensorFlow dependencies
66-
"org.tensorflow" % "libtensorflow" % targetTensorFlowVersion,
67-
"org.tensorflow" % "libtensorflow_jni" % targetTensorFlowVersion
66+
"org.tensorflow" % "tensorflow" % targetTensorFlowVersion
6867
)
6968

7069
lazy val testDependencies = Seq(

0 commit comments

Comments
 (0)