TensorFlow-GPU环境配置之四——配置和编译TensorFlow
Posted Conor白
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了TensorFlow-GPU环境配置之四——配置和编译TensorFlow相关的知识，希望对你有一定的参考价值。
首先，使用configure进行配置
配置完成后，使用bazel编译retrain命令，编译命令中加入--config=cuda即为启用GPU
编译进行中。。。
编译完成
编译完成后，调用retrain命令训练时，会显示显卡信息
以上是关于TensorFlow-GPU环境配置之四——配置和编译TensorFlow的主要内容，如果未能解决你的问题，请参考以下文章
python 深度学习环境安装（tensorflow-gpu）
环境配置（近期实测）——Ubuntu16.04+CUDA9.0+tensorflow-gpu填坑记
conda配置安装pytorch tensorflow-gpu