一、ROCm：一個可移植、高性能的GPU計算平臺

隨著AI時代的來臨，顯卡的作用不再僅僅只局限于游戲，特別是在Stable Diffusion火爆全網(wǎng)之后，越來越多的玩家將顯卡當(dāng)成AIGC生產(chǎn)力工具。

只不過，Stable Diffusion的早期版本對AMD顯卡支持不太友好，給大家造成了只有N卡才適合玩AI的錯覺。

可能很多同學(xué)不知道的是，AMD的Radeon GPU也是能夠做深度學(xué)習(xí)的，而且性能也并不差。

去年12月份，AMD推出的ROCm 6.0開發(fā)平臺進(jìn)一步優(yōu)化了RX 7000系列顯卡的AI性能，特別是Stable Diffusion的出圖效率更是獲得了翻倍提升。

這里先解釋一下什么是AMD ROCm！

ROCm （Radeon Open Compute Platform ）是 AMD 基于開源項目的 GPU計算生態(tài)系統(tǒng)，類似于 NVIDIA 的 CUDA。

ROCm 支持多種編程語言、編譯器、庫和工具，以加速科學(xué)計算、自動駕駛、人工智能和機(jī)器學(xué)習(xí)等領(lǐng)域的應(yīng)用。

ROCm還支持多種加速器廠商和架構(gòu)，提供了開放的可移植性和互操作性。

目前深度學(xué)習(xí)開發(fā)一般都是使用Python，而大數(shù)據(jù)平臺往往都會部署在Linux操作系統(tǒng)之上。因此，AMD ROCm目前主要精力放在Linux系統(tǒng)上，Windows端則還有待完善。

得益于雙發(fā)射設(shè)計的流處理器單元，RX 7900 XT僅憑5376個流處理器就達(dá)到了52TFOPS的單精度浮點運(yùn)算性能，而擁有10240個流處理器的RTX 4080 Super浮點性能同樣則是55TFOPS。

至于RX 7900 XTX，其單精度浮點性能則接近60TFOPS。

Stable Diffusion主要依賴GPU的單精度浮點性能以及顯存容量與帶寬，這兩方面RX 7900 XT都不遜于RTX 4080 SUPER，甚至顯存容量還更大一些。

現(xiàn)在有了AMD ROCm 6.0的加持，終于可以發(fā)揮出這張頂級顯卡強(qiáng)大的AI算力。

二、Stable Diffusion出圖速度對比：Linux系統(tǒng)下出圖效率近乎翻倍

測試平臺如下：

本次我們主要對比RX 7900 XT/XTX在Ubuntu與windows系統(tǒng)下的出圖效率，看看AMD ROCm到底能帶來多大提升。

1、RX 7900 XT對比測試

首先是Windows 11系統(tǒng)，打開Stable Diffusion之后，我們設(shè)置DPN++ 2M Karras采樣、迭代步數(shù)50、Euler a采樣、1024x768分辨率、CFG為7、總批次10，單批數(shù)量1。

在Windows系統(tǒng)下，RX 7900 XT基于DirectML加速模式生成10張圖共計耗時9分51秒，也就是591秒。

進(jìn)入Ubuntu 22.04.3 LTS系統(tǒng)，設(shè)置好ROCm之后，同樣的參數(shù)與模型生成10張圖只用掉了5分05秒，即305秒。

對比之下，在Ubuntu 22.04.3 LTS系統(tǒng)下，RX 7900 XT的出圖效率相比Windows系統(tǒng)高了足足94%。

2、RX 7900 XTX對比測試

使用同樣的參數(shù)來測試RX 7900 XTX顯卡。

在Windows系統(tǒng)下，RX 7900 XTX基于DirectML加速模式生成10張圖總計耗時9分18秒，也就是558秒。

進(jìn)入Ubuntu 22.04.3 LTS系統(tǒng)，設(shè)置好ROCm之后，同樣的參數(shù)與模型生成10張圖只用掉了4分45秒，即285秒。

在Ubuntu 22.04.3 LTS系統(tǒng)下，RX 7900 XTX的出圖效率相比Windows系統(tǒng)領(lǐng)先了96%，也就是近乎翻倍的性能提升。

三、小結(jié)：高性價比的AIGC顯卡

目前一張非公RX 7900 XT售價在5299元左右，對位的非公RTX 4080 Super售價則是8599元，N卡貴了60%。其實在游戲性能方面，這兩款顯卡并沒有太大差距。

現(xiàn)在AMD也意識到了AI應(yīng)用對于GPU的重要性，加大了對AMD ROCm平臺的投入，經(jīng)過不斷的更新，如今AMD ROCm 6.0在AI算力方面的提升早已超出了大多數(shù)玩家的認(rèn)知。

在我們的測試中，RX 7900 XT在Linux系統(tǒng)下使用Stable Diffusion ROCm 6.0版時，其出圖效率幾乎是Windows系統(tǒng)的2倍。再加上55TFOPS的單精度浮點運(yùn)算性能以及20GB GDDR6X超大容量顯存，對于有較高專業(yè)AIGC應(yīng)用需求的專業(yè)用戶而言，在有限的預(yù)算下，RX 7900 XT可以說是非常適合的選擇。

當(dāng)然，也希望AMD能夠繼續(xù)優(yōu)化ROCm在Windows系統(tǒng)下的性能表現(xiàn)，讓更多的AMD用戶能擁有屬于自己的高性價比AIGC顯卡。

       原文標(biāo)題 : ROCm加持出圖效率翻倍！AMD RX 7900 XT Linux AI性能體驗