tensorflow學習筆記(三)

2021-09-25 19:23:14 字數 808 閱讀 4283

1.是否列印裝置分配日誌

sess = tf.session(config = tf.configproto(log_device_placement = true))
2.如果指定的裝置不存在,是否允許tf自動分配裝置

sess = tf.session(config = tf.configproto(allow_soft_placement = true))
3.手工指派裝置

with tf.device('/cpu:0'):

...

建立乙個裝置環境,這個環境下的operation都統一執行在環境指定的裝置上

4.控制gpu資源使用率

(1)開始時分配少量的gpu容量,然後按需增加,由於不會釋放記憶體,所以會導致碎片

config = tf.configproto()

config.gpu_options.allow_growth = true

sess = tf.session(config=config, ...)

(2)設定每個gpu應該拿出多少容量給程序使用,0.4代表40%

gpu_options = tf.gpuoptions(per_process_gpu_memory_fraction=0.4)

config = tf.configproto(gpu_options=gpu_options)

sess = tf.session(config=config,...)

tensorflow學習筆記三

分布式tensorflow就是多台伺服器參加乙個tensorflow圖的分布式執行,分布式我感覺就是原來在一台計算機上面執行好幾個程序這些程序互動是由os控制的,而分布式就是把這些程序放在了不同的機器上面執行,他們之間的互動是由分布式框架控制的,實際分布式的核心或者說基本點還是執行的程序。一提到分布...

tensorflow學習筆記

tensorflow安裝可以直接通過命令列或者原始碼安裝,在此介紹tensorflow8命令列安裝如下 安裝tensorflow sudo pip install upgrade 另外,解除安裝tensorflow命令為 sudo pip uninstall tensorflow tensorflo...

Tensorflow學習筆記

1.如何在虛擬機器中安裝tensor flow 1 首先安裝pip pip install 2 pip install 2.學習tensorflow需要學習 python and linux 3.使用 tensorflow,你必須明白 tensorflow 1 使用圖 graph 來表示計算任務.2...