added fp16 flag

PipelineAI · Dec 11, 2017 · 166a902 · 166a902
1 parent 4cb476d
commit 166a902
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/gpu.ml/Dockerfile.cpu b/gpu.ml/Dockerfile.cpu
@@ -1,6 +1,9 @@
 #FROM fluxcapacitor/package-tensorflow-7a7fe93-4c0052d-full-cpu:master
 FROM fluxcapacitor/package-tensorflow-16d39e9-d690fdd-full-cpu:master
 
+# ADD THIS!
+# TF_FP16_MATMUL_USE_FP32_COMPUTE=0 
+
 WORKDIR /root
 
 # Based on the following:  https://github.com/jupyterhub/jupyterhub/blob/master/Dockerfile

diff --git a/gpu.ml/Dockerfile.gpu b/gpu.ml/Dockerfile.gpu
@@ -1,5 +1,8 @@
 FROM fluxcapacitor/package-tensorflow-16d39e9-d690fdd-full-gpu:master
 
+# ADD THIS!
+# TF_FP16_MATMUL_USE_FP32_COMPUTE=0
+
 WORKDIR /root
 
 # Based on the following:  https://github.com/jupyterhub/jupyterhub/blob/master/Dockerfile