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