tensorflow2 0視訊記憶體設定

遇到乙個問題：

新買顯示卡視訊記憶體8g但是tensorflow執行的時候介面顯示只有約6.3g的視訊記憶體可用，如下圖：

即限制了我的視訊記憶體，具體原因為什麼我也不知道，但原來的視訊記憶體小一些的顯示卡就沒有這個問題。

目前的解決辦法是：

官方文件解決

然後對應的中文部落格

總結一下，就是下面的兩個辦法：

設定每次使用時按需申請

# 獲取所有gpu組成list

gpus = tf.config.experimental.list_physical_devices(
'gpu'
)# 設定按需申請
# 由於我這裡僅有一塊gpu,multi-gpu需要for一下
tf.config.experimental.set_memory_growth(gpu[0]
,true
)

gpus = tf.config.experimental.list_physical_devices(

'gpu'
)# 對需要進行限制的gpu進行設定
tf.config.experimental.set_virtual_device_configuration(gpus[0]
,[tf.config.experimental.virtualdeviceconfiguration(memory_limit=
8192)]
)

如何設定為全域性設定，即每次開啟都不用再新增這幾行**就為上次的設定，

我目前還沒發現。。。

如有看到的請教我一下嘛

Tensorflow2 0 啟用函式

常用啟用函式及對應特點 神經網路結構的輸出為所有輸入的加權和，這導致整個神經網路是乙個線性模型。而線性模型不能解決異或問題，且面對多分類問題，也顯得束手無策。所以為了解決非線性的分類或回歸問題，啟用函式必須是非線性函式。神經網路中啟用函式的主要作用是提供網路的非線性建模能力。這是因為反向傳播演算法就...

初步了解TensorFlow2 0

為什麼要學習tensorflow？深度學習能夠更好地抽取資料中的規律，從而給公司帶來更大的價值 tensorflow是強大且靈活的開源框架 使用廣泛 2.0更加強大 易用 成熟 tensorflow是什麼？是google的開源軟體庫 採用資料流圖，用於數值計算 支援多平台 gpu cpu 移動裝置 ...

Tensorflow2 0簡單應用 一

我是初學者 參考 1.匯入tf.keras tensorflow2推薦使用keras構建網路，常見的神經網路都包含在keras.layer中 最新的tf.keras的版本可能和keras不同 import tensorflow as tf from tensorflow.keras import l...