常聽人說AI對存儲性能提出了新需求，那么AI對存儲有怎樣的需求呢？

人工智能全流程中，涉及數(shù)據(jù)的采集、數(shù)據(jù)的預(yù)處理、模型訓(xùn)練、模型部署等環(huán)節(jié)，跟以往操作差異最大的在于訓(xùn)練環(huán)節(jié)，為此，有外媒動手做了一個評測。結(jié)果顯示，在訓(xùn)練環(huán)節(jié)，高性能存儲設(shè)備對于性能的提升完全沒有幫助。

測試架構(gòu)是這樣的，一臺戴爾PowerEdge R740xd服務(wù)器，有兩塊至強(qiáng)黃金6130 CPU，256GB DRAM內(nèi)存，用它來運(yùn)行byteLAKE的AI測試，測試的變量是三款SSD，一個是鎧俠SAS 口的PX04S，一個是三星的983ZET，還有一個是英特爾傲騰900P。

鎧俠SAS口 PX04S SSD

三星PCIe卡983ZET

英特爾傲騰900P

測試中分析了機(jī)器學(xué)習(xí)在訓(xùn)練階段的性能表現(xiàn)，測試程序用的是byteLAKE的EWA Gurad，它是一個實(shí)時監(jiān)測模型，該模型包括一個輸入層，22個卷積層，5個池化層，2個路由層，還有一個探測層。

測試優(yōu)劣的評判標(biāo)準(zhǔn)是時間，就是做5000個Epoch所耗費(fèi)的時間，每換一次SSD就進(jìn)行一次測試，測試結(jié)果如下。

鎧俠 98小時 24分

三星 98小時 44分

英特爾 98小時 42分

明顯可見，本地的SSD對性能幾乎沒有影響，換了不同性能的SSD，既有SAS口的，也有PCIe口的，既有NAND SSD也有機(jī)遇3D Xpoint的SSD，最終性能表現(xiàn)都差不多。

在數(shù)據(jù)的獲取時，SSD會有優(yōu)勢，但在計算環(huán)節(jié)，對SSD的要求并不高，那是不是說明CPU的訓(xùn)練能力不夠，導(dǎo)致瓶頸呢？

為了進(jìn)一步測試，在服務(wù)器里加入來了一塊英偉達(dá)T4顯卡，然后再跑一下測試。

鎧俠 4h 30分鐘

三星 4小時 28分鐘

Intel 4小時 27分鐘

GPU的加速效果還是非常明顯的，提升了22倍，計算性能提升來的話，SSD的性能差異是否明顯呢？從數(shù)據(jù)來看，顯然還是沒什么影響。

測試結(jié)果發(fā)現(xiàn)，高性能的存儲并沒有提升AI訓(xùn)練階段的性能，由于AI模型的復(fù)雜結(jié)構(gòu)限制，機(jī)器學(xué)習(xí)訓(xùn)練的時間要遠(yuǎn)遠(yuǎn)高于數(shù)據(jù)載入的時間。雖然英偉達(dá)T4顯卡加速了機(jī)器學(xué)習(xí)的速度，但并沒有讓SSD發(fā)揮優(yōu)勢，

當(dāng)然，想要看具體某個零部件對于系統(tǒng)性能影響的話還需要許多細(xì)致的工作，比如，換一個AI框架，換別的零部件什么的。這一測試其實(shí)是有意義的，可以讓我們找出有利于性能提升的部分，把錢花在刀刃上。
責(zé)任編輯:pj