tensorflow 2.x訓練時，預設狀態下gpu視訊記憶體只能分配82%左右（筆者電腦為rtx 2070 max-q 8gb，windows 10，cuda 10.1，cudnn 7.6，tensorflow 2.1.0）。
很多部落格給出如下的**：
gpus = tf.config.experimental.list_physical_devices('gpu')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, true)
以下**可以解決：
gpus = tf.config.experimental.list_physical_devices('gpu')
tf.config.experimental.set_virtual_device_configuration(
gpus[0], [tf.config.experimental.virtualdeviceconfiguration(memory_limit=8192)])
tensorflow基礎使用2
fetch和 feed的用法 1.fetch的用法也就是在session執行時，可以以列表的方式傳參，從而同時run多個op coding utf 8 import tensorflow as tf how to use fetch input1 tf.constant 3.0 input2 tf....
Tensorflow學習筆記2
參考 1 classification分類學習 from tensorflow.examples.tutorials.mnist import input data mnist input data.read data sets mnist data one hot true mnist庫是手寫體數...
tensorflow入門筆記（2）
在tensorflow中，變數 tf.variable 的作用是儲存和更新神經網路中的引數。一般來說，權重引數 weights 賦予隨機初始值最為常見。偏置項常常用常數來設定初始值 import tensorflow as tf weights為乙個2 3的矩陣，初始值為正態分佈隨機數，標準差而2，...