1. 程式人生 > >TensorFlow-GPU環境配置之四——配置和編譯TensorFlow

TensorFlow-GPU環境配置之四——配置和編譯TensorFlow

http mage body class 調用 技術 圖片 環境 進行

首先,使用configure進行配置

技術分享圖片

配置完成後,使用bazel編譯retrain命令,編譯命令中加入--config=cuda即為啟用GPU

技術分享圖片

編譯進行中。。。

技術分享圖片

編譯完成

技術分享圖片

編譯完成後,調用retrain命令訓練時,會顯示顯卡信息

技術分享圖片

TensorFlow-GPU環境配置之四——配置和編譯TensorFlow