1. 程式人生 > >tensorflow相關配置

tensorflow相關配置

<tensorflow實戰google深度學習框架>

都使用GPU進行訓練

​ 沒執行一個程式時使用設定不同的 CUDA_VISIBLE_DEVICES ,在執行程式前先執行export CUDA_VISIBLE_DEVICES = 0,表示指定編號為0的GPU給程式,然後執行程式。在要執行第二個程式的使用 export CUDA_VISIBLE_DEVICES = 1指定另外一個GPU給程式

GPU和CPU同時訓練

​ 在想讓程式只在CPU上執行時,使用export CUDA_VISIBLE_DEVICES ="" 指定空,這樣執行的程式就只在 CPU中,而不會影響GPU