Added TFoS version

luigiba · Oct 8, 2019 · ba1d69c · ba1d69c
1 parent ec0f762
commit ba1d69c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 
 # OpenKEonSpark  
-This is a distributed version of the framework OpenKE (https://github.com/thunlp/OpenKE) using the library TensorFlowOnSpark (https://github.com/yahoo/TensorFlowOnSpark).  
+This is a distributed version of the framework OpenKE (https://github.com/thunlp/OpenKE) using the library TensorFlowOnSpark (https://github.com/yahoo/TensorFlowOnSpark version 1.4.4).  
 Please refer to https://github.com/tensorflow/examples/blob/master/community/en/docs/deploy/distributed.md to have an overview about how Distributed Tensorflow training works.
 
 

diff --git a/colab/commands.txt b/colab/commands.txt
@@ -9,7 +9,7 @@ os.environ["JAVA_HOME"] = "/usr/lib/jvm/java-8-openjdk-amd64"
 !sudo apt-get install scala
 !wget -q https://archive.apache.org/dist/spark/spark-2.1.1/spark-2.1.1-bin-hadoop2.7.tgz 
 !tar xf spark-2.1.1-bin-hadoop2.7.tgz
-!python3 -m pip install tensorflowonspark py4j pyspark
+!python3 -m pip install tensorflowonspark==1.4.4 py4j pyspark
 
 #connect Google Drive
 from google.colab import drive