tensorflow學習筆記1 GPU配置

2021-09-03 01:45:28 字數 393 閱讀 2287

import tensorflow as tf

config = tf.configproto()

config.gpu_options.allow_growth = true

config.gpu_options.per_process_gpu_memory_fraction = 0.7

config.gpu_options.allocator_type = 'bfc'

關於bfc演算法:

將記憶體分塊管理,按塊進行空間分配和釋放。

通過split操作將大記憶體塊分解成使用者需要的小記憶體塊。

通過merge操作合併小的記憶體塊,做到記憶體碎片**

通過bin這個抽象資料結構實現對空閒塊高效管理。

tensorflow學習筆記1

在跑minist demo時,遇到了這幾句 batchsize 6 label tf.expand dims tf.constant 0,2,3,6,7,9 1 index tf.expand dims tf.range 0,batchsize 1 concated tf.concat 1,inde...

TensorFlow學習筆記1

1 tensorflow 谷歌第二代人工智慧學習系統 2 tensorflow顧名思義tensor flow。tensor的意思是 張量,flow的意思是 流動,合起來就是 張量的流動 3 系統架構及程式設計模型。其中系統架構如圖1所示,程式設計模型如圖2所示。圖1 tensorflow系統架構圖 ...

TensorFlow學習筆記1

編寫tensorflow的兩個步驟 構建計算圖graph 使用session去執行graph中的operation 這裡寫描述 三個基本概念 rank rank一般是指資料的維度,其與線性代數中的rank不是乙個概念。其常 用rank舉例如下。shape 指tensor每個維度資料的個數,可以用py...