英偉達(dá)A800芯片也要被禁售了？中國(guó)AI大模型還能怎么玩？國(guó)產(chǎn)GPU能否挑大梁

電子發(fā)燒友網(wǎng)報(bào)道（文/李彎彎）近日消息，美國(guó)考慮進(jìn)一步限制向中國(guó)出口AI芯片，商務(wù)部最早可能在下個(gè)月初采取行動(dòng)。在沒(méi)有獲得許可證的情況下，美商務(wù)部將禁止英偉達(dá)等制造商向中國(guó)客戶運(yùn)送AI芯片。英偉達(dá)專供中國(guó)的A800芯片，在無(wú)許可證的情況下也將被禁售。

A800是英偉達(dá)專為中國(guó)客戶推出的產(chǎn)品，用以解決美國(guó)商務(wù)部的半導(dǎo)體出口新規(guī)，取代A100 GPU，該產(chǎn)品在2022年第三季度起投產(chǎn)。近段時(shí)間，大模型迅猛發(fā)展，國(guó)內(nèi)在新規(guī)之前采購(gòu)的可流通A100并不多，目前各互聯(lián)網(wǎng)廠商爭(zhēng)相采購(gòu)的主要是A800。

A800被禁售，國(guó)內(nèi)AI大模型訓(xùn)練受阻？

在AI大模型的訓(xùn)練上，現(xiàn)在基本上是用英偉達(dá)的A100、H100以及特供中國(guó)的減配版 A800、H800。2020年和2022年，英偉達(dá)先后推出了A100芯片、H100芯片，其中A100顯卡又提供80GB、40GB兩個(gè)顯存版本，H100則提供80GB顯存。

A100擁有540億晶體管，采用臺(tái)積電7nm工藝制程，支持FP16、FP32和FP64浮點(diǎn)運(yùn)算，為人工智能、數(shù)據(jù)分析和HPC數(shù)據(jù)中心等提供算力。

H100集成800億晶體管，采用臺(tái)積電定制的4nm工藝，相比于A100，H100在FP16、FP32和FP64計(jì)算上比A100快三倍，非常適用于當(dāng)下流行且訓(xùn)練難度高的大模型。

2022年9月，美國(guó)商務(wù)部對(duì)GPU 的出口規(guī)定，主要限制是顯卡的算力和帶寬，算力上線是4800 TOPS，帶寬上線是600 GB/s。A800的帶寬為400GB/s，低于A100的600GB/s，H800的參數(shù)并未公布，據(jù)知情人士透露只有H100（900 GB/s）到約一半。

這意味著A800、H800在進(jìn)行AI模型訓(xùn)練的時(shí)候，需要耗費(fèi)更長(zhǎng)的時(shí)間，不過(guò)這相比于其他的同類產(chǎn)品，性能仍然是高的。

最近大半年時(shí)間，隨著ChatGPT的出圈，國(guó)內(nèi)外各大互聯(lián)網(wǎng)公司、AI企業(yè)都在大力研發(fā)自己的AI大模型產(chǎn)品，而這也造成了對(duì)GPU的極大需求。各大互聯(lián)網(wǎng)廠商都在大力采購(gòu)英偉達(dá)的GPU芯片，包括百度、阿里、騰訊、字節(jié)跳動(dòng)等等。

在國(guó)內(nèi)，還可流通的A100并不多，據(jù)一家服務(wù)器定制企業(yè)的銷售經(jīng)理描述，需求太過(guò)火爆，甚至有人從機(jī)房里把芯片禁運(yùn)政策生效前購(gòu)買的A100 80GB拆出來(lái)倒賣。

如今，國(guó)內(nèi)市場(chǎng)還是主要倚賴在政策允許范圍內(nèi)的A800芯片。不過(guò)大模型火爆之后，因?yàn)樾枨筇?，A800也出現(xiàn)供貨周期拉長(zhǎng)、價(jià)格上漲的情況，據(jù)業(yè)內(nèi)人士透露，之前兩周時(shí)間就能到貨，現(xiàn)在樂(lè)觀的情況也要4-6周的時(shí)間，甚至更長(zhǎng)。

如今，根據(jù)美國(guó)商務(wù)部的意思，A800在無(wú)許可證的情況下將被禁售，這意味著，后續(xù)國(guó)內(nèi)企業(yè)在大模型的研究中，不僅僅是買到A800需要更長(zhǎng)時(shí)間的問(wèn)題，而是很可能買不到的問(wèn)題。而且不僅僅是A800，其他AI芯片向中國(guó)的銷售也可能受阻，包括原本也可以用于一些模型訓(xùn)練，性能較差一些的V100等。這意味著，國(guó)內(nèi)企業(yè)AI大模型后續(xù)的訓(xùn)練將受阻。

國(guó)產(chǎn)GPU能否挑大梁？

如此看來(lái)，國(guó)內(nèi)AI后續(xù)的發(fā)展，它對(duì)算力的需求似乎就只能指望國(guó)產(chǎn)GPU了。目前國(guó)產(chǎn)GPU的情況如何呢？

大模型訓(xùn)練需要處理高顆粒度的信息，對(duì)云端訓(xùn)練芯片的芯片處理信息的精細(xì)度和算力速度要求更高，現(xiàn)階段國(guó)產(chǎn)GPU大多數(shù)不具備支撐大模型訓(xùn)練所需的能力。

不同于多媒體和圖形處理的單精度浮點(diǎn)計(jì)算（FP32）計(jì)算需求，在超算領(lǐng)域，雙精度浮點(diǎn)計(jì)算能力FP64是進(jìn)行高算力計(jì)算的硬性指標(biāo)。

目前國(guó)內(nèi)GPU芯片的云端訓(xùn)練公司，大多只能處理單精度浮點(diǎn)計(jì)算，如壁仞科技（通用GPU芯片BR100）、天數(shù)智芯（“智鎧100”）、寒武紀(jì)（云端推理思元270）的產(chǎn)品在FP32的理論指標(biāo)上做得不錯(cuò)，但沒(méi)有處理FP64的能力。

所以說(shuō)，從短期來(lái)看，國(guó)產(chǎn)的GPU產(chǎn)品還難以支撐起AI大模型的訓(xùn)練，這意味著英偉達(dá)A800芯片被禁售，國(guó)內(nèi)AI大模型接下來(lái)的研究將難以為繼。

那是不是說(shuō)，無(wú)論是國(guó)內(nèi)的AI大模型企業(yè)，還是國(guó)產(chǎn)的GPU企業(yè)，真的就束手無(wú)策了？也并不是。據(jù)業(yè)內(nèi)人士介紹，對(duì)于算力的需求主要與模型參數(shù)有關(guān)，大廠現(xiàn)在所做的通用大模型依然只能使用英偉達(dá)的產(chǎn)品。但是，垂類行業(yè)的大模型（金融、醫(yī)療等）或者參數(shù)量不像GPT那樣大的模型，一些國(guó)產(chǎn)GPU也能夠做到。

比如，國(guó)內(nèi)某短視頻平臺(tái)參數(shù)量約為1000億的大模型，就用了寒武紀(jì)的產(chǎn)品，做訓(xùn)練和推理。當(dāng)然該模型訓(xùn)練使用了英偉達(dá)的A100。

此外，今年6月10日，天數(shù)智芯對(duì)外宣布，在天垓100加速卡的算力集群，基于北京智源人工智能研究院70億參數(shù)的Aquila語(yǔ)言基礎(chǔ)模型，使用代碼數(shù)據(jù)進(jìn)行繼續(xù)訓(xùn)練，穩(wěn)定運(yùn)行19天，模型收斂效果符合預(yù)期，證明天數(shù)智芯有支持百億級(jí)參數(shù)大模型訓(xùn)練的能力。

摩爾學(xué)院院長(zhǎng)李豐此前對(duì)媒體表示，公司已經(jīng)可以支持3億參數(shù)量模型的訓(xùn)練。

首都在線表示，公司與燧原科技聯(lián)合開發(fā)的AIGC模型方案已發(fā)布內(nèi)測(cè)，產(chǎn)品可以從容應(yīng)對(duì)前期投入大、算力要求高、算法模型快速迭代創(chuàng)新的需求，并廣泛支持文本、語(yǔ)音、視覺(jué)等各技術(shù)方向的模型訓(xùn)練。

可見，目前國(guó)產(chǎn)GPU雖然在參數(shù)量較大的模型訓(xùn)練上存在不足，但是確實(shí)已經(jīng)能夠在一些中小參數(shù)模型的訓(xùn)練上有所作為。

登臨科技聯(lián)合創(chuàng)始人王平此前對(duì)媒體表示，相比于國(guó)際領(lǐng)先的GPU產(chǎn)品，國(guó)產(chǎn)GPU確實(shí)在滿足大模型訓(xùn)練上存在差距，因?yàn)榇饲靶袠I(yè)在產(chǎn)品定義上未朝大模型方向做設(shè)計(jì)。

而現(xiàn)在國(guó)產(chǎn)GPU公司都在朝著大模型方向去布局，相信未來(lái)終能夠找到方向，進(jìn)行突破。而且如今A800在無(wú)許可證的情況下如果禁售，這對(duì)于國(guó)產(chǎn)GPU的發(fā)展將會(huì)有強(qiáng)大的刺激作用，因?yàn)槿绻煌黄疲馕吨鴩?guó)內(nèi)未來(lái)AI大模型的發(fā)展將無(wú)計(jì)可施，這似乎成了唯一的出路。

目前業(yè)界正在做相關(guān)的探索，比如，思考能否通過(guò)Chiplet、先進(jìn)封裝的方式提高算力。研究如何提升芯片與芯片間的數(shù)據(jù)傳輸速度等。昆侖芯就表示，相較于第一代產(chǎn)品，昆侖芯2在算力、互聯(lián)上都有大幅優(yōu)化，公司正在不斷研發(fā)新的產(chǎn)品和技術(shù)，以滿足ChatGPT等大模型的應(yīng)用需求。

小結(jié)

很顯然，在A100、H100之后，若A800在無(wú)許可證下被禁售，意味著國(guó)內(nèi)AI大模型后續(xù)的訓(xùn)練將無(wú)GPU可用。目前而言國(guó)產(chǎn)GPU幾乎還不能用于較大規(guī)模參數(shù)模型的訓(xùn)練。不過(guò)可以看到，一些產(chǎn)品在較小規(guī)模，比如幾億、甚至百億級(jí)別參數(shù)規(guī)模的模型訓(xùn)練上，已經(jīng)可以有所作為。而且目前眾廠商都在朝著大模型應(yīng)用方向進(jìn)行探索，從長(zhǎng)遠(yuǎn)來(lái)看，必然會(huì)逐步取得進(jìn)展。