tensorflow 如何控制使用cpu數量

tensorflow讀取資料有時候非常占用資源，針對這一情況，我們可以通過控制程式使用cpu的數量來限制

步驟如下：

1、bashrc 中修改指定個數

$ export cpu_num=2

cpu_num = int(os.environ.get('cpu_num',1))

config = tf.configproto(device_count=,
inter_op_parallelism_threads = cpu_num,
intra_op_parallelism_threads = cpu_num,
log_device_placement=true)
with tf.session(config = config) as sess:

深度學習 TensorFlow 入門之基本使用

使用 tensorflow,你必須明白 tensorflow 賦值或者從其中獲取資料.乙個 tensorflow 圖描述了計算的過程.為了進行計算,圖必須在 會話 裡被啟動.會話 將圖的 op 分發到諸如 cpu 或 gpu 之類的 裝置 上,同時提供執行 op 的方法.這些方法執行後,將產生的 t...

學習筆記 TensorFlow 入門之基本使用

使用 tensorflow,你必須明白 tensorflow 賦值或者從其中獲取資料.乙個 tensorflow 圖描述了計算的過程.為了進行計算,圖必須在 會話 裡被啟動.會話 將圖的 op 分發到諸如 cpu 或 gpu 之類的 裝置 上,同時提供執行 op 的方法.這些方法執行後,將產生的 t...

學習筆記 TensorFlow 入門之基本使用

使用 tensorflow,你必須明白 tensorflow 賦值或者從其中獲取資料.乙個 tensorflow 圖描述了計算的過程.為了進行計算,圖必須在 會話 裡被啟動.會話 將圖的 op 分發到諸如 cpu 或 gpu 之類的 裝置 上,同時提供執行 op 的方法.這些方法執行後,將產生的 t...