使用SSE4.2和AVX重新编译Tensorflow

时间:2018-04-15 19:39:53

标签: python-3.x ubuntu tensorflow

我跟随着pyimagesearch书,它附带了一个ubuntu 16.04虚拟机,已经安装了所有库(sklearn,keras等),并且设置了所有内容。但是,当我第一次使用keras运行某些​​代码时,我收到了警告The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computation,而AVX也是如此。我查看了this question以及installing from source,但两人似乎都认为尚未安装TensorFlow,它已经安装了。我只需要卸载并重新开始吗?或者我还能做些什么吗?

1 个答案:

答案 0 :(得分:0)

这只是一个警告,它不会阻止你使用TensorFlow。但是如果您希望TensorFlow使用这些说明,您需要重新安装它。 TensorFlow 1.6+ pip packages are compiled with AVX by default,但是对于其他说明,您需要从源代码编译TensorFlow。