ARMcpu
编辑环境变量
sudo gedit ~/.bashrc
- 在最后一行添加
export OPENBLAS_CORETYPE=ARMV8
- 最后,激活环境变量
source ~/.bashrc
- 即可!
X86
首先要确定的是:如果你使用的是虚拟机,要确定是否支持avx指令集!
cat /proc/cpuinfo #下图中没有avx,则不支持。
- 降级Tensorflow版本到1.5或更低。(不推荐,但可以尝试一下。)
pip install --upgrade tensorflow==1.5
2. tensorflow源码编译的方法。(不推荐,太麻烦了)
3. 换个CPU 。(不推荐,有钱任性也可以)
4. 使用conda的环境解决。(推荐,好用)
conda create -n tensorflow python=3.6 # 创建虚拟环境,已经是虚拟环境就忽略吧
source activate tensorflow # 激活虚拟环境
conda install tensorflow-gpu == 1.14.0 # 安装tensorflow