首先,使用configure进行配置

TensorFlow-GPU环境配置之四——配置和编译TensorFlow

配置完成后,使用bazel编译retrain命令,编译命令中加入--config=cuda即为启用GPU

TensorFlow-GPU环境配置之四——配置和编译TensorFlow

编译进行中。。。

TensorFlow-GPU环境配置之四——配置和编译TensorFlow

编译完成

TensorFlow-GPU环境配置之四——配置和编译TensorFlow

编译完成后,调用retrain命令训练时,会显示显卡信息

TensorFlow-GPU环境配置之四——配置和编译TensorFlow

相关文章: