TensorFlow:从源重建已安装的TensorFlow会导致任何内部问题吗?

时间:2017-03-06 05:29:39

标签: tensorflow deep-learning

我目前正在AWS P2实例上运行TensorFlow培训,该实例已安装tensorflow 0.12。我使用pip将其升级到1.0,但如果我想从源配置它并使用bazel重建它以提高CPU计算速度(根据我运行TF时给出的建议),这可能吗?或者我是否必须从头开始卸载TF并重建?te

1 个答案:

答案 0 :(得分:1)

为了使用您的CPU潜力,您应该使用正确的标志进行编译并重新安装。

遵循本指南:https://www.tensorflow.org/install/install_sources

我用,具体案例:

bazel build --config=opt --copt=-msse4.2 --copt=-msse4.1 --copt=-msse3 //tensorflow/tools/pip_package:build_pip_package

希望它有所帮助。