首先,使用configure进行配置
配置完成后,使用bazel编译retrain命令,编译命令中加入--config=cuda即为启用GPU
编译进行中。。。
编译完成
编译完成后,调用retrain命令训练时,会显示显卡信息
标签: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