平台:ubuntu16.04 LTS
环境:cuda9.1 tensorflow-gpu1.4 nvidia390.25 anconda3.0
问题:cuda配置成功,tensorflow也能够code,但是使用的却是cpu加速,
log:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2.
解决方法:使用conda list 检查已经安装的包,发现除了tensorflow-gpu之外,还有tensorflow 1.5。猜测可能是因为使用了cpu版本的tensorflow。于是卸载tensorflow,但是并没有起到效果。
尝试:conda uninstall protobuf,卸载全部tensorflow相关包,然后conda install tensorflow-gpu。随后编译提示切换设备为gpu0,成功使用gpu加速。
提示:keras和tensorflow不能共存。在conda install keras的时候会一起安装tensorflow,可能会导致上述问题。