tensorflow 1 13原始碼編譯

2021-10-02 22:03:39 字數 492 閱讀 9878

bazel和tensorflow有嚴格的對應版本,對應關係如下:

編譯tensorflow之前先要安裝bazel;

編譯tensorflow

$ git clone

$ cd tensorflow

$ git checkout r1.13  # r1.9, r1.10, etc.

(這裡選用的

r1.13)

$ ./configure

在configure時,其他選項選擇no,cuda選項選擇yes,configure成功之後,然後編譯gpu版本的tensorflow。

$ bazel build --config=opt --config=cuda //tensorflow/tools/graph_transforms:transform_graph

參考:

113 TensorFlow變數集合

乙個tensorflow程式斷開的部分可能要建立變數 如果有一種方法來訪問所有的變數是非常有用的 因為這個原因tensorflow提供了集合,是一些張量的集合 或者是其他的物件,就像tf.variable 例項一樣 預設情況下 tf.variable 物件被放置在下面的兩個集合中 tf.graphk...

AbstractCollection原始碼分析

abstractcollection抽象類提供了collection的骨架實現,collection分析請看 這裡直接看它的 是如何實現的.public abstract iterator iterator 該方法沒有實現.public abstract int size 該方法沒有實現.publi...

ThreadPoolExecutor原始碼閱讀

執行緒池解決兩個問題 一是復用執行緒,減少建立銷毀執行緒帶來系統開銷 二是限定系統資源使用邊界,避免大量執行緒消耗盡系統記憶體 適用於互不依賴,執行時間短,不需要對執行緒控制操作的執行緒 新增任務時,1.若執行緒數量小於corepoolsize,則新增執行緒執行任務 2.若執行緒數量大於等於core...