File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -41,6 +41,11 @@ ENV PATH $SPARK_HOME/bin:$PATH
4141ENV PYTHONPATH /opt/spark/python/lib/py4j-0.10.7-src.zip:/opt/spark/python/lib/pyspark.zip:$PYTHONPATH
4242ENV 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.
4550VOLUME /mnt/tensorframes
4651WORKDIR /mnt/tensorframes
Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments