无法运行简单的Tensorflow程序

时间:2018-02-07 15:10:51

标签: ubuntu tensorflow bazel tensorflow-serving

每当我尝试运行一个简单的tensorflow程序时,例如:

import tensorflow as tf 

a = tf.constant([4,3], name = "const_a")    
b = tf.reduce_sum(a, name = "add_b")    
c = tf.reduce_prod(b, name = "mul_c")
d = tf.multiply(b,c, name = "mul_d")

sess = tf.Session()
output = sess.run(d)
writer = tf.summary.FileWriter('./my_graphs', sess.graph) 

我收到以下错误:

I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA

我尝试安装bazel,但他们不支持Ubuntu 17.04(如网站上提到的支持Ubuntu Linux平台:16.04(LTS)14.04(LTS))

我也收到了一个错误:

the build command is only supported from within a workspace

每当我尝试运行bazel安装的代码时  bazel build -c opt --copt=-march=native --copt=-mfpmath=both --config=cuda -k //tensorflow/tools/pip_package:build_pip_package

如何在没有任何警告和错误的情况下使用Tensorflow?

1 个答案:

答案 0 :(得分:0)

这不是错误;这是一个警告。

如果您不想看到它,请将其放在TF代码的顶部。

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'