tensorflow-gpu源码编译

本文转自：https://blog.csdn.net/caroline_wendy/article/details/80868120

(1)  ./configure

Please specify the location of python. [Default is /data2/wcl1/tensorflow/venv/bin/python] ## 选择Python的版本，2或3

## 其余选N或n

Do you wish to build TensorFlow with CUDA support? [y/N]: y  # 选择GPU版本，y
CUDA support will be enabled for TensorFlow.

Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 9.0]: 8.0  # CUDA版本，与服务器一致，8.0

Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7.0]: 6.0  # cuDNN版本，与服务器一致，6.0

## 其余选N，或默认

Do you want to use clang as CUDA compiler? [y/N]: N  ## 选择nvcc
nvcc will be used as CUDA compiler.

## 其余选N，或默认

Configuration finished

扫描二维码关注公众号，回复： 5430786 查看本文章

(2)  bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package