码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-02-12 11:11:07      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:src   gpo   flow   技术分享   显示   gpu   训练   conf   配置   

首先,使用configure进行配置

技术分享图片

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

技术分享图片

编译进行中。。。

技术分享图片

编译完成

技术分享图片

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

技术分享图片

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

标签:src   gpo   flow   技术分享   显示   gpu   训练   conf   配置   

原文地址:https://www.cnblogs.com/conorpai/p/8443704.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!