當前位置: 華文世界 > 科技

打造AI創作畫圖「煉丹爐」 高效能固態硬碟成數據遷移提效關鍵

2024-02-22科技

2023年,以Stable Diffusion為代表的AIGC(AI Generated Content,生成式AI)創作工具橫空出世,它利用深度學習和擴散模型演算法,不僅拓寬了藝術創作的邊界,而且簡單的互動形式,即使沒有專業繪畫技能的人也能輕松創造出高質素的照片、插畫等藝術作品,一時間在全球掀起了「煉丹」的熱潮。

不過,我們在享受AIGC帶來的便利的同時,也要直面它所存在的一些問題,例如版權以及無題材限制的創作,會讓普通使用者混淆視聽,「有圖有真相」甚至會成為假新聞的幫兇,例如2023年3月國外社交媒體上釋出的美國前總統杜林普從「被捕」--「越獄」--「火並」事件,簡直活脫脫的一部好萊塢劇本大片,但這些都是看熱鬧不嫌事大的網友們用AIGC「即興」炮製出來的。

不過AIGC最直接的問題,就是本地儲存資源消耗日益凸顯,不斷更新的模型、外掛程式以及使用者積累的生成素材內容,大幅占用硬碟的儲存空間。如果使用者想要在新電腦或者新硬碟中使用自己習慣的AIGC,還要將這些數據「原原本本」的遷移過來,而這些高達數萬個甚至十幾萬個的檔對於固態硬碟來說將成為一項不小的挑戰。

以我們訓練了幾個月的Stable Diffusion為例,目前的檔體積達到了160GB,總數量達到了個;生成的圖片大小雖然只有8.46GB,但是數量高達10000張以上,如果是重度的「煉丹」發燒友甚至是職業的「方士」,這個數量級將更為龐大。要遷移這些數據,固態硬碟至少要TB級別。如果再考慮到AIGC的升級叠代,可以考慮2TB、4TB等大容量產品。

但Stable Diffusion資料夾中的相當一部份數量的檔都是只有幾個KB大小的細碎檔。我們知道,固態硬碟的參數大家最關註的就是順序讀寫速度,也就是說只要檔大小不超過SLC緩存的容量,就可以超高速傳輸。而這些細碎檔,則更考驗的是固態硬碟的隨機效能,隨機讀寫效能越強,處理零碎檔(如啟動程式、遊戲等)任務就越快。

那麽,遷移Stable Diffusion檔和普通的檔傳輸有何區別?我們找來了威騰電子旗下高效能品牌WD_BLACK的旗艦選手WD_BLACK SN850X NVMe SSD 2TB來簡單測試一下。這款固態硬碟采用原廠3D NAND快閃記憶體顆粒和效能強大的主控芯片,還配備了2GB容量的DDR4獨立緩存芯片,最大連續讀寫效能高達7300MB/s和6600MB/s,可以說是WD_BLACK的最強「黑盤」。

我們將160GB的Stable Diffusion遷移到威騰電子WD_BLACK SN850X 2TB中,傳輸完成用時大約2分49秒,傳輸速度圖形如下,可知傳輸速度最高速度均達到3.xGB/s,最低速度也只有十幾到幾十MB/s,但這個數據量顯然沒有超出SLC緩存的容量(空盤時大約580GB),這就是大檔和大量細碎檔同時傳輸的結果。

接下來將9個共計176GB的壓縮檔傳輸到威騰電子WD_BLACK SN850X 2TB中,用時僅為50秒左右,從傳輸圖形可看到基本全程維持在3.2GB/s以上。由此可見Stable Diffusion遷移對於固態硬碟的壓力還比單獨大檔傳輸要大很多,而配備獨立的DRAM緩存的產品在隨機效能方面更有優勢,應對這些大量的細碎檔的讀寫效率比DRAMLess產品更高一些,所以作業系統OS磁碟機也是推薦配備獨立DRAM緩存的產品。

有人說,在未來取代人類工作的不是AI,而是使用AI的人。縱然我們離這個時代還有些距離,但AI繪畫的普及不僅讓創作超越人們的想象,也極大降低了創意實作技術門檻。但「煉丹爐」的打造方面,除了高算力的顯卡,選擇一款高容量、高速度的固態硬碟也是提升效率的關鍵,例如威騰電子WD_BLACK SN850X這類產品,為AI創作帶來更好、更快的使用體驗,助力創作者更自由地探索無盡的創意世界。

威騰電子(WD)2TB大容量 筆記本SSD固態硬碟 M.2 SN850X PCIe4.0 2280 NVMe 遊戲電競電腦擴充套件
¥1349 領券減20
京東
購買