提到TensorFlow很多人第一反應(yīng)是谷歌出品的深度學(xué)習(xí)框架工業(yè)界最成熟的選擇但真到自己動(dòng)手裝環(huán)境、寫模型、調(diào)參數(shù)的時(shí)候往往又會(huì)覺得這一腳踩下去水很深。我最初接觸TensorFlow是在1.x時(shí)代被session、placeholder、graph這些概念折騰得夠嗆后來2.x出來之后順手多了但很多老教程還是1.x的寫法導(dǎo)致新手照著抄都報(bào)錯(cuò)。這篇博客我想從一個(gè)實(shí)際項(xiàng)目使用者的角度把TensorFlow的安裝、核心概念、建模流程、踩坑經(jīng)驗(yàn)以及2024年它在與PyTorch的競爭中所處的生態(tài)位都掰開揉碎講一遍。如果你正準(zhǔn)備入門深度學(xué)習(xí)或者已經(jīng)在用其他框架想對(duì)比一下TensorFlow這篇文章應(yīng)該能幫你少走很多彎路。1. 為什么還要聊TensorFlow先搞清楚它能做什么1.1 TensorFlow到底是什么它在解決什么問題TensorFlow本質(zhì)是一個(gè)數(shù)值計(jì)算庫核心邏輯是構(gòu)建數(shù)據(jù)流圖把復(fù)雜的數(shù)學(xué)運(yùn)算拆成一個(gè)個(gè)節(jié)點(diǎn)操作節(jié)點(diǎn)之間通過張量Tensor傳遞數(shù)據(jù)然后讓框架自動(dòng)完成求導(dǎo)、并行計(jì)算、分布式部署這些臟活累活。你可以把它理解成一個(gè)超級(jí)計(jì)算器——只不過這個(gè)計(jì)算器不僅能算加減乘除還能算神經(jīng)網(wǎng)絡(luò)里動(dòng)輒上億個(gè)參數(shù)的梯度并且能在GPU、CPU甚至多臺(tái)服務(wù)器上同時(shí)干活。這個(gè)定位決定了TensorFlow最核心的適用場(chǎng)景深度神經(jīng)網(wǎng)絡(luò)的訓(xùn)練與推理。從圖像分類、目標(biāo)檢測(cè)到自然語言處理、推薦系統(tǒng)幾乎你能想到的主流AI應(yīng)用都可以用TensorFlow搭起來。它尤其適合那些需要把模型產(chǎn)品化的公司——訓(xùn)練好的模型可以轉(zhuǎn)成SavedModel格式通過TensorFlow Serving部署到服務(wù)器或者用TensorFlow Lite部署到手機(jī)和嵌入式設(shè)備上。這也是即使PyTorch在研究圈越來越流行TensorFlow在工業(yè)界仍然有大量存量系統(tǒng)的原因。1.2 誰適合學(xué)TensorFlow誰可以先繞道如果你是想快速做實(shí)驗(yàn)、發(fā)論文、基于已有模型做二次開發(fā)那么PyTorch的調(diào)試體驗(yàn)確實(shí)更友好目前學(xué)術(shù)界的多數(shù)新模型也首選PyTorch。但如果你面臨下面幾種情況TensorFlow會(huì)更加合適一是公司已有TensorFlow的模型庫和部署鏈路需要維護(hù)和迭代二是你要做大規(guī)模分布式訓(xùn)練TensorFlow的分布式策略在工程上更成熟三是你要做端側(cè)部署TensorFlow Lite和TFLite Micro在移動(dòng)端和MCU上有完整的工具鏈。當(dāng)然如果你是純新手想通過一個(gè)框架弄懂深度學(xué)習(xí)的核心概念TensorFlow 2.x配合Keras這套高層API也足夠友好心理負(fù)擔(dān)可以降下來。我的建議是別被框架之爭帶偏至少在入門階段TensorFlow和PyTorch的底層原理高度相似學(xué)會(huì)一個(gè)遷移到另一個(gè)只是語法層面的熟練問題。關(guān)鍵是先動(dòng)手把模型跑起來理解張量、梯度、優(yōu)化器這些核心概念。2. TensorFlow安裝實(shí)操從零到能跑通第一個(gè)模型2.1 安裝前的關(guān)鍵決策版本、硬件與Python環(huán)境TensorFlow的安裝看似簡單——一行pip install tensorflow——但實(shí)際動(dòng)手時(shí)很多人踩的第一個(gè)坑就是版本與硬件不匹配。在2024年這個(gè)時(shí)間點(diǎn)官方穩(wěn)定版已經(jīng)是2.16左右注意2.x的API跟1.x差別極大如果你在網(wǎng)上搜到2019年之前的教程里面大概率還是tf.Session()這類舊寫法直接抄必然會(huì)報(bào)錯(cuò)。安裝前先確認(rèn)三件事Python版本TensorFlow 2.16要求Python 3.9~3.12更老的3.7、3.8雖然還能裝但可能裝到的是舊版本不值得。建議直接用Python 3.10或3.11。是否用GPU如果電腦有NVIDIA顯卡且顯存不低于4GB建議裝GPU版。TensorFlow 2.x的pip包tensorflow已經(jīng)默認(rèn)包含GPU支持不需要單獨(dú)裝tensorflow-gpu。前提是安裝好CUDA和cuDNN或者直接裝tensorflow[and-cuda]讓pip幫你拉依賴。虛擬環(huán)境千萬別圖省事用全局Python直接裝。我見過太多人把系統(tǒng)Python裝壞了最后只能重裝系統(tǒng)。用venv或者conda單獨(dú)建一個(gè)環(huán)境TensorFlow的依賴比如numpy、protobuf跟其他深度學(xué)習(xí)庫、數(shù)據(jù)處理庫很容易互相打架虛擬環(huán)境是必須的。2.2 從零安裝的完整流程CPU版和GPU版這里給出一個(gè)我在干凈機(jī)器上實(shí)測(cè)過的安裝流程適配Windows/Linux/macOSmacOS的GPU支持受限一般用CPU版。第一步建虛擬環(huán)境以conda為例conda create -n tf python3.11 -y conda activate tf第二步安裝TensorFlowCPU版pip install tensorflowGPU版推薦用官方推薦的捆綁安裝pip install tensorflow[and-cuda]如果你更習(xí)慣自己管理CUDA可以走傳統(tǒng)路線先裝CUDA 11.x或12.x再裝cuDNN 8.x然后pip install tensorflow。但這個(gè)傳統(tǒng)路線非常容易遇到版本不匹配的問題——我自己曾經(jīng)因?yàn)镃UDA 12.2和TensorFlow編譯時(shí)用的12.0不完全一致折騰了一整天。后來發(fā)現(xiàn)直接pip install tensorflow[and-cuda]最省事它會(huì)自動(dòng)安裝匹配的CUDA運(yùn)行庫和cuDNN雖然下載體積大接近3GB但勝在省心。第三步驗(yàn)證安裝import tensorflow as tf print(tf.__version__) print(tf.config.list_physical_devices(GPU))CPU版會(huì)打印出版本號(hào)GPU版如果配置正確會(huì)打印出類似[PhysicalDevice(name/physical_device:GPU:0, device_typeGPU)]的信息。注意如果你看到的是空列表說明GPU沒被識(shí)別最常見原因是驅(qū)動(dòng)版本太老先升級(jí)NVIDIA驅(qū)動(dòng)。2.3 安裝后的常見地雷圓圈進(jìn)度條與V2兼容問題裝完TensorFlow有個(gè)很反直覺的現(xiàn)象第一次執(zhí)行import tensorflow時(shí)有時(shí)候沒有報(bào)錯(cuò)但命令行會(huì)持續(xù)輸出各種INFO日志甚至看起來像卡住了。這時(shí)候別慌TensorFlow在初始化時(shí)要做很多檢查慢是正常的。但如果你每次導(dǎo)入都慢得離譜多半是CPU指令集兼容問題。蘋果M系列芯片的用戶建議安裝tensorflow-macos或者直接使用支持Metal的TensorFlow版本否則性能會(huì)差很多。還有一個(gè)高頻報(bào)錯(cuò)是AttributeError: module tensorflow has no attribute Session。出現(xiàn)這個(gè)就說明你用的還是2.x版本但代碼是1.x寫的。解決辦法是改用2.x的Keras接口或者兼容運(yùn)行tf.compat.v1.disable_eager_execution()但后者不推薦——除非你在維護(hù)老項(xiàng)目否則別在2024年學(xué)舊API。3. 核心概念與實(shí)操要點(diǎn)張量、自動(dòng)微分和Keras高層API3.1 張量到底是什么跟數(shù)組和矩陣有啥關(guān)系TensorFlow名字里這個(gè)詞Tensor就是模型的血液。你可以把張量簡單理解為多維數(shù)組的統(tǒng)稱0維張量是標(biāo)量一個(gè)數(shù)1維張量是向量一列數(shù)2維張量是矩陣一個(gè)表格3維及以上就是更高維的數(shù)據(jù)塊。圖片在深度學(xué)習(xí)中就是一個(gè)典型的4維張量形狀是[batch_size, height, width, channels]比如一批32張256x256的RGB圖片形狀就是[32, 256, 256, 3]。為什么特意強(qiáng)調(diào)張量而不是數(shù)組因?yàn)樵谏疃葘W(xué)習(xí)里張量不僅有數(shù)值還伴隨著數(shù)據(jù)類型float32、int32等、形狀shape和計(jì)算圖上的依賴關(guān)系。你對(duì)張量做運(yùn)算TensorFlow會(huì)自動(dòng)記錄整個(gè)計(jì)算鏈路這樣后面反向傳播求梯度時(shí)它才能沿著鏈路把誤差一層層傳回去。這就像記賬時(shí)候的溯源系統(tǒng)——每一筆錢從哪來、到哪去都有跡可循梯度才能準(zhǔn)確分配到每個(gè)參數(shù)頭上。實(shí)操上常用的幾個(gè)張量操作我列在下面tf.constant()創(chuàng)建不可變張量適合存固定數(shù)據(jù)。tf.Variable()創(chuàng)建可訓(xùn)練變量模型的權(quán)重和偏置都用它。tf.reshape()/tf.transpose()改變張量形狀或調(diào)換維度順序處理數(shù)據(jù)必經(jīng)之路。tf.cast()強(qiáng)制類型轉(zhuǎn)換比如把float64轉(zhuǎn)成float32減少顯存占用。tf.squeeze()/tf.expand_dims()去掉或增加長度為1的維度做數(shù)據(jù)對(duì)齊時(shí)特別好用。新手最容易犯的錯(cuò)誤是對(duì)張量的形狀沒有直覺。比如全連接層的輸入要求二維[batch, features]你給進(jìn)去一個(gè)一維數(shù)組它就會(huì)報(bào)shape不匹配的錯(cuò)。我的經(jīng)驗(yàn)是每次把數(shù)據(jù)喂給模型之前先檢查data.shape心里默念一遍幾個(gè)維度、每個(gè)維度多大能避免90%的維度坑。3.2 自動(dòng)微分框架幫你把導(dǎo)數(shù)算得明明白白傳統(tǒng)的機(jī)器學(xué)習(xí)要手動(dòng)推導(dǎo)梯度公式再寫代碼實(shí)現(xiàn)。神經(jīng)網(wǎng)絡(luò)層數(shù)一多推導(dǎo)過程簡直能讓人崩潰。TensorFlow的自動(dòng)微分autodiff把這個(gè)過程完全自動(dòng)化了你只需要定義前向計(jì)算過程框架會(huì)利用鏈?zhǔn)椒▌t自動(dòng)構(gòu)建反向傳播所需的梯度計(jì)算圖。這就是tf.GradientTape做的事。一個(gè)最經(jīng)典的例子定義一個(gè)變量x計(jì)算y x^2然后求y對(duì)x的導(dǎo)數(shù)2ximport tensorflow as tf x tf.Variable(3.0) with tf.GradientTape() as tape: y x ** 2 grad tape.gradient(y, x) print(grad.numpy()) # 輸出 6.0注意兩點(diǎn)第一GradientTape要放在前向計(jì)算的外面它就像一臺(tái)錄像機(jī)錄制里面的所有運(yùn)算過程第二默認(rèn)情況下tape用過一次就會(huì)被釋放如果需要多次求梯度要設(shè)置persistentTrue。實(shí)際訓(xùn)練模型時(shí)你不需要手動(dòng)寫梯度更新的邏輯optimizer.apply_gradients()會(huì)幫你把梯度應(yīng)用到可訓(xùn)練變量上。但你理解了GradientTape的原理就能看懂訓(xùn)練循環(huán)到底在干什么遇到loss不下降的時(shí)候也知道往哪個(gè)方向排查。3.3 Keras就是你的模型積木工廠TensorFlow 2.x把Keras作為官方高層API目的就是讓用戶不用再跟底層計(jì)算圖細(xì)節(jié)死磕。Keras提供了三種構(gòu)建模型的方式我根據(jù)項(xiàng)目復(fù)雜度給你建議第一種Sequential順序模型。適合層與層之間直線堆疊的簡單網(wǎng)絡(luò)比如一個(gè)只有全連接層和激活層的MLPmodel tf.keras.Sequential([ tf.keras.layers.Dense(128, activationrelu), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10, activationsoftmax) ])第二種Functional函數(shù)式模型。適合有分支、有合并的復(fù)雜網(wǎng)絡(luò)比如多輸入模型、殘差連接。你需要自己定義輸入張量并串聯(lián)各個(gè)層inputs tf.keras.Input(shape(28, 28)) x tf.keras.layers.Flatten()(inputs) x tf.keras.layers.Dense(64, activationrelu)(x) outputs tf.keras.layers.Dense(10, activationsoftmax)(x) model tf.keras.Model(inputsinputs, outputsoutputs)第三種Subclassing子類化。通過繼承tf.keras.Model來自定義前向傳播邏輯適合研究性項(xiàng)目。自由度最高但調(diào)試成本也高我不建議新手一開始就用。我個(gè)人的做法是80%的項(xiàng)目用Sequential或Functional都能搞定只有需要魔改模型內(nèi)部結(jié)構(gòu)時(shí)才用Subclassing。千萬別為了炫技搞復(fù)雜建模方式Keras已經(jīng)足夠強(qiáng)大。4. 實(shí)戰(zhàn)案例用TensorFlow訓(xùn)練一個(gè)手寫數(shù)字識(shí)別模型4.1 數(shù)據(jù)準(zhǔn)備從張量到Dataset理論說再多不如跑一個(gè)真實(shí)模型。我們以MNIST數(shù)字識(shí)別為例——這是深度學(xué)習(xí)界的Hello World。數(shù)據(jù)直接用Keras自帶的數(shù)據(jù)集不用額外下載import tensorflow as tf # 加載數(shù)據(jù)第一次會(huì)自動(dòng)下載 (x_train, y_train), (x_test, y_test) tf.keras.datasets.mnist.load_data() # 歸一化把像素值從0~255縮放到0~1加速收斂 x_train x_train.astype(float32) / 255.0 x_test x_test.astype(float32) / 255.0 # 增加通道維度變成 [batch, 28, 28, 1] x_train x_train[..., tf.newaxis] x_test x_test[..., tf.newaxis] # 使用Dataset構(gòu)建輸入管道 train_ds tf.data.Dataset.from_tensor_slices((x_train, y_train)) train_ds train_ds.shuffle(10000).batch(64) test_ds tf.data.Dataset.from_tensor_slices((x_test, y_test)).batch(64)這里有幾個(gè)實(shí)操細(xì)節(jié)歸一化一定要做這相當(dāng)于把所有特征放到同一個(gè)量綱下否則梯度更新會(huì)非常不穩(wěn)定用tf.data.Dataset而不是直接把數(shù)組喂給模型是為了在大數(shù)據(jù)量下能夠做預(yù)取、亂序和并行處理避免訓(xùn)練時(shí)CPU/GPU數(shù)據(jù)吞吐不匹配。4.2 構(gòu)建模型與訓(xùn)練配置我們用最簡單的多層感知機(jī)MLP來做分類。輸入是28x28的灰度圖先通過Flatten層拉平成784維向量然后接兩個(gè)全連接層model tf.keras.Sequential([ tf.keras.layers.Flatten(input_shape(28, 28, 1)), tf.keras.layers.Dense(128, activationrelu), tf.keras.layers.Dense(64, activationrelu), tf.keras.layers.Dense(10, activationsoftmax) ]) model.compile( optimizeradam, losssparse_categorical_crossentropy, metrics[accuracy] ) history model.fit( train_ds, validation_datatest_ds, epochs10 )為什么損失函數(shù)用sparse_categorical_crossentropy因?yàn)槲覀兊臉?biāo)簽是整數(shù)0~9而不是one-hot編碼的10維向量。如果標(biāo)簽是one-hot就用categorical_crossentropy。這個(gè)區(qū)分是新手經(jīng)常搞糊涂的地方一句話記牢整數(shù)標(biāo)簽用sparse_前綴獨(dú)熱編碼不用。訓(xùn)練過程會(huì)輸出每一輪的loss和accuracy大概10輪之后測(cè)試集準(zhǔn)確率能達(dá)到98%以上。如果你在CPU上跑每輪也就十幾秒能非常直觀地感受到模型在收斂。4.3 模型評(píng)估與導(dǎo)出部署訓(xùn)練完成后先看一眼在測(cè)試集上的表現(xiàn)loss, acc model.evaluate(test_ds) print(f測(cè)試集準(zhǔn)確率: {acc:.4f})接下來用模型預(yù)測(cè)單張圖片predictions model.predict(x_test[:1]) predicted_class tf.argmax(predictions, axis-1).numpy() print(predicted_class)最后導(dǎo)出為SavedModel格式方便部署model.save(mnist_model)這個(gè)mnist_model文件夾里就是完整的模型定義和權(quán)重。部署到生產(chǎn)環(huán)境時(shí)用TensorFlow Serving加載它即可也可以用Python的tf.saved_model.load()來加載做推理?？偟膩碚f從構(gòu)建到部署的鏈路非常順暢這也是TensorFlow的看家本領(lǐng)。5. TensorFlow與PyTorch的流行趨勢(shì)2024年到底該怎么選5.1 兩邊的生態(tài)現(xiàn)狀科研向左工業(yè)向右2024年學(xué)術(shù)界的使用習(xí)慣已經(jīng)明顯偏向PyTorch——新發(fā)布的論文、預(yù)訓(xùn)練模型、開源代碼絕大多數(shù)都是PyTorch版本。這背后的直接原因是PyTorch的調(diào)試體驗(yàn)更接近Python直覺print張量值不用手動(dòng)跑會(huì)話想斷點(diǎn)調(diào)試就斷點(diǎn)。而TensorFlow 2.x雖然已經(jīng)默認(rèn)啟用了Eager Execution很多老用戶的習(xí)慣和記憶還停留在1.x的不友好階段導(dǎo)致它在口耳相傳中吃虧。但工業(yè)界完全是另一套邏輯。我接觸過不少做推薦系統(tǒng)、廣告CTR預(yù)估、風(fēng)控模型的公司線上留存的核心模型依然是TensorFlow。為什么一是基礎(chǔ)設(shè)施沉淀公司多年前就圍繞TensorFlow搭建了特征工程、模型訓(xùn)練、模型上線、AB測(cè)試的完整數(shù)據(jù)鏈路遷移成本極高二是TensorFlow Serving的成熟度領(lǐng)先支持模型熱更新、多版本管理、高并發(fā)請(qǐng)求這些在企業(yè)級(jí)場(chǎng)景非常關(guān)鍵。PyTorch雖然有TorchServe但部署生態(tài)的穩(wěn)定性和團(tuán)隊(duì)熟悉度仍有差距。5.2 2024年值得關(guān)注的新動(dòng)勢(shì)融合與互補(bǔ)從2024年看兩個(gè)框架的流行趨勢(shì)不再是你死我活而是邊界融合。PyTorch推出了TorchScript和LibTorch努力在部署側(cè)補(bǔ)課TensorFlow這邊則把重心放在JAX兼容、Keras 3多后端支持上。Keras 3是個(gè)重要信號(hào)——它已經(jīng)支持PyTorch和JAX作為后端也就是說你可以用Keras的高層API但底層引擎換成PyTorch。這意味著什么對(duì)開發(fā)者來說框架的鎖定效應(yīng)在減弱。你今天用TensorFlow Keras寫好的模型未來完全可以切換到JAX后端跑研究實(shí)驗(yàn)?zāi)憬裉煊肞yTorch訓(xùn)練出的權(quán)重也有工具可以轉(zhuǎn)成TensorFlow的格式部署。我的建議是選框架沒那么重要重要的是掌握深度學(xué)習(xí)的基礎(chǔ)概念和工程化思維。具體到落地決策我給自己定了幾條原則供你參考如果做純研究、發(fā)論文、復(fù)現(xiàn)最新模型優(yōu)先PyTorch。如果做企業(yè)級(jí)應(yīng)用、考慮長期維護(hù)和上線部署TensorFlow依然穩(wěn)妥。如果團(tuán)隊(duì)已經(jīng)熟悉某個(gè)框架別輕易換工具服務(wù)于項(xiàng)目。如果處于學(xué)習(xí)階段選一個(gè)深入學(xué)透兩個(gè)都接觸一下不要在東張西望中浪費(fèi)時(shí)間。6. 常見問題與排查技巧實(shí)錄6.1 訓(xùn)練速度慢到懷疑人生怎么辦很多人寫的模型在GPU上跑不起來一看任務(wù)管理器GPU占用率為0那問題多半出在數(shù)據(jù)管道上。tf.data.Dataset默認(rèn)的讀取方式是單線程順序加載如果你的數(shù)據(jù)預(yù)處理邏輯重比如圖片解碼、隨機(jī)增強(qiáng)CPU會(huì)成為瓶頸。解決方案是加上預(yù)取和并行處理train_ds train_ds.shuffle(10000).batch(64).prefetch(tf.data.AUTOTUNE)prefetch(tf.data.AUTOTUNE)會(huì)讓數(shù)據(jù)加載和后端訓(xùn)練并行進(jìn)行效果立竿見影。另外如果圖片很多還可以用map的時(shí)候指定num_parallel_callstf.data.AUTOTUNE。還有一個(gè)我碰到過很多次的情況代碼沒問題但訓(xùn)練期間顯存占用持續(xù)飆升最后OOM。多半是batch_size設(shè)太大或者模型里某層用了過大的特征圖?？梢韵日{(diào)小batch_size驗(yàn)證一下再把輸入圖片分辨率降一檔基本就能解決。6.2 Loss不降和NaN的經(jīng)典排查路徑Loss一開始就很低或者一動(dòng)不動(dòng)最典型的原因就是模型的輸出層跟損失函數(shù)不匹配。比如二分類問題輸出層用了softmax加2個(gè)節(jié)點(diǎn)但損失函數(shù)選了binary_crossentropy——這就是經(jīng)典的模型結(jié)構(gòu)沒跟損失對(duì)上的錯(cuò)誤。修正辦法要么輸出層改1個(gè)節(jié)點(diǎn)配sigmoid要么輸出層保持2個(gè)節(jié)點(diǎn)配categorical_crossentropy。Loss變成NaN則基本是數(shù)值不穩(wěn)定。常見原因有學(xué)習(xí)率過大、輸入數(shù)據(jù)里包含NaN、梯度爆炸。排查步驟我一般這樣走檢查輸入數(shù)據(jù)是否有無窮值或NaNtf.debugging.check_numerics(data, data)。把學(xué)習(xí)率降低一個(gè)數(shù)量級(jí)比如從0.01降到0.001重訓(xùn)一次。在模型里加BatchNormalization或ClipByNorm控制梯度模長。6.3 跨版本遷移的兼容性雷區(qū)如果你在老項(xiàng)目上使用TensorFlow可能遇到tf.contrib、tf.app.run等1.x特有模塊。這些模塊在2.x中已經(jīng)被移除沒有直接替換的對(duì)應(yīng)物。我的建議是不要試圖兼容干脆按2.x的Keras API重寫成本通常低于預(yù)期。如果實(shí)在需要跑舊代碼可以用tf.compat.v1模塊但要開啟tf.compat.v1.disable_eager_execution()并且把tf.contrib手動(dòng)替換掉——這個(gè)過程非常折磨人非必要不走這條路。另外tf.placeholder在2.x中已經(jīng)不存在了改用函數(shù)式API的定義方式。tf.Session()也沒有了直接用Python函數(shù)就能在前向計(jì)算。記住這點(diǎn)就能避免大部分教程過時(shí)導(dǎo)致的報(bào)錯(cuò)。6.4 內(nèi)存泄漏的觀察方向訓(xùn)練很多輪之后越來越卡內(nèi)存逐漸漲滿這通常是數(shù)據(jù)管道的迭代器沒有正確釋放或者在自定義訓(xùn)練循環(huán)里創(chuàng)建了大量tf.Variable但沒被垃圾回收。輕量級(jí)解決辦法是每輪訓(xùn)練結(jié)束后調(diào)用gc.collect()并把重復(fù)創(chuàng)建的模型對(duì)象換成單例如果用了tf.data.Dataset確保迭代器只保留當(dāng)前批次不要保留整個(gè)數(shù)據(jù)集的迭代狀態(tài)。真正在跑大數(shù)據(jù)時(shí)建議用tf.keras.utils.Sequence做數(shù)據(jù)生成器它對(duì)內(nèi)存的管理更明確。7. 寫在最后我這兩年使用TensorFlow的真實(shí)體會(huì)說句掏心窩的話TensorFlow帶給我的感覺一直很矛盾它一方面有厚重的歷史包袱各種版本割裂令人抓狂另一方面它的工程化能力又確實(shí)是一眾深度學(xué)習(xí)框架里最扎實(shí)的。我個(gè)人在過去兩年里從TensorFlow 1.x遷移到2.x又用Keras 3嘗試對(duì)接JAX后端最大的感受是框架的迭代速度遠(yuǎn)比我們想象中快今天糾結(jié)的選哪個(gè)框架可能明年就變成一個(gè)無關(guān)緊要的問題。如果你還在猶豫怎么入門我的建議是別把精力花在對(duì)比框架的優(yōu)劣上直接選定一個(gè)搭好環(huán)境跑通一個(gè)小模型然后逐漸加大難度。第一次跑通MNIST那會(huì)兒的興奮感我相信你很快就會(huì)體驗(yàn)到。當(dāng)你真正理解了張量、梯度、優(yōu)化器這些核心概念會(huì)發(fā)現(xiàn)TensorFlow和PyTorch之間的差異不過就是語法糖層面的差別。最后分享一個(gè)實(shí)用小技巧無論用哪個(gè)框架都要養(yǎng)成最小化可復(fù)現(xiàn)實(shí)驗(yàn)的習(xí)慣。遇到bug時(shí)把問題化簡到盡可能小的規(guī)模比如用一個(gè)只有幾條數(shù)據(jù)的小數(shù)組去復(fù)現(xiàn)在Google Colab上快速驗(yàn)證思路這樣能大幅縮短排查時(shí)間。深度學(xué)習(xí)項(xiàng)目90%的時(shí)間可能都花在數(shù)據(jù)、參數(shù)和調(diào)試上框架本身反而只是最小的一部分。祝你在TensorFlow的世界里玩得順模型一跑就收斂。