win10 + CUDA10 + tensorflow1.12 GPU版本编译安装

参考链接

Windows 10 环境下编译 Tensorflow 1.12 GPU 版本指南

教程已经很详细了，参考操作基本上没问题。

这里记录一下注意事项：

1. Python我直接装了个windows 3.5.4版本，没有使用Anaconda。需要安装下面的包

pip3 install six numpy wheel
pip3 install keras_applications==1.0.5 --no-deps
pip3 install keras_preprocessing==1.0.3 --no-deps

2. Bazel使用的0.18版本
3. CUDA版本
   CUDA：10.1
   cuDNN：7.5.0 for CUDA 10.1
4. Eigen补丁必须打，否则显卡算力大于6的会报错
5. 我是在msys2里面编译的，需要配置以下环境变量

export MSYS_NO_PATHCONV=1
export MSYS2_ARG_CONV_EXCL="*"
export PATH="/d/bazel:$PATH"
export PATH="/d/python/python35:/d/python/python35/Scripts:$PATH"
export BAZEL_SH=E:/msys64/usr/bin/bash.exe
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/bin:$PATH"
export PATH="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/extras/CUPTI/libx64:$PATH"

BAZEL_SH必须要配置，不然使用bazel时报错

6. 编译命令

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package ./tmp/tensorflow_pkg

最后贴一个我的测试截图