参考链接
Windows 10 环境下编译 Tensorflow 1.12 GPU 版本指南
教程已经很详细了,参考操作基本上没问题。
这里记录一下注意事项:
- 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
- Bazel使用的0.18版本
- CUDA版本
CUDA:10.1
cuDNN:7.5.0 for CUDA 10.1 - Eigen补丁必须打,否则显卡算力大于6的会报错
- 我是在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时报错
- 编译命令
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