我們在捷智算平臺(tái)上收到的最常見問題之一是如何充分利用我的 GPU？今天針對這個(gè)問題，為大家做一個(gè)詳細(xì)的解答，同時(shí)分享給大家合適的 GPU 選擇指南。
最重要的因素
首先，你需要有足夠的 VRAM 來適應(yīng)你的模型。這應(yīng)該是你首先要考慮的。
Tensor Core 用于矩陣乘法，這對于任何 AI 應(yīng)用都至關(guān)重要。從 2017 年的 V100 開始，專業(yè) GPU 中引入了 Tensor Core，2018 年的 GeForce 20 系列也引入了消費(fèi)級(jí) GPU。捷智算平臺(tái)上的絕大多數(shù) GPU 都配有 Tensor Core，那么問題就在于有多少個(gè)呢？
事實(shí)上，不是。Tensor Core 非常快 - 快到大多數(shù)時(shí)候都沒有得到充分利用，具體取決于批處理大小。因此，瓶頸通常是帶寬，包括內(nèi)存和互連（GPU 之間）。
FP16 與 FP8
FP8 用 8 位表示數(shù)字，而不是之前標(biāo)準(zhǔn)的 FP16 的 16 位。最新的 GPU（例如 H100 和即將推出的 B100）支持針對 FP8 的優(yōu)化。
精度較低時(shí)，模型可以使用較低的 VRAM 和帶寬運(yùn)行，因此計(jì)算量也較少。但是，您必須小心，不要顯著降低模型性能。如果您已經(jīng)測試過可以從 FP8 中受益，或者有理由認(rèn)為可以受益，請確保您使用的是支持 FP8 的 GPU。
最佳推理精選
消費(fèi)級(jí)顯卡的性價(jià)比通常更高，因?yàn)槠髽I(yè)級(jí) GPU 的標(biāo)價(jià)明顯更高。大多數(shù)云提供商不提供消費(fèi)級(jí) GPU，但捷智算平臺(tái)提供。
如果您可以將模型裝入 24 GB VRAM 卡中，那么 3090 和 4090 的推理價(jià)值將非常高。超過 24 GB，您將被迫進(jìn)入企業(yè)領(lǐng)域，而價(jià)值卻無法接近。如果您尚未確定特定模型，那么將其裝入 24 GB VRAM 卡中將大大降低您的推理成本。
除此之外，下一個(gè)價(jià)值等級(jí)是較低級(jí)別的企業(yè)卡，例如 V100 32 GB、A6000（48 GB）、L40（48 GB）和 6000 Ada（48 GB）。
然后，你終于有了 A100 80 GB 和 H100。如果你需要這么高的性能來進(jìn)行推理，你肯定會(huì)知道的。總的來說，A100 提供了更好的推理價(jià)值，而 H100 最適合訓(xùn)練。
企業(yè) GPU
1、擴(kuò)展計(jì)算：NVLink 和 InfiniBand
搭載 NVLink 的 NVIDIA A100
由于 GPU 必須互連且受帶寬限制，因此將服務(wù)器從 1 個(gè) GPU 擴(kuò)展到 8 個(gè) GPU 并不會(huì)自動(dòng)實(shí)現(xiàn) 8 倍性能?，F(xiàn)代 GPU 支持更快的互連，因此對于 V100 等較舊的 GPU，這種下降會(huì)更加明顯。
在捷智算平臺(tái)上，如果您租用 2x、4x 或 8x H100/A100 服務(wù)器，您的 VM 會(huì)在可用時(shí)自動(dòng)通過 NVLink 互連。
NVLink 連接服務(wù)器內(nèi)的 GPU，而 InfiniBand 連接多臺(tái)服務(wù)器。預(yù)計(jì) InfiniBand 的價(jià)格會(huì)略有上漲，約為 5-10%。請記住，除非您正在訓(xùn)練非常大的模型，否則這并不是必需的。
對于推理，我們已經(jīng)看到，運(yùn)行每個(gè)虛擬機(jī) 4 個(gè) GPU 且每個(gè) 8xH100 節(jié)點(diǎn)運(yùn)行兩個(gè)實(shí)例的開源模型可獲得最佳性能，優(yōu)于運(yùn)行 2 個(gè) GPU 或 8 個(gè) GPU 的模型。此外，我們已經(jīng)看到，使用 Triton 推理的 TensorRT-LLM 的吞吐量大約是 vLLM 的兩倍。
2、SXM 與 PCIe（僅限 H100、A100 和 V100）
NVIDIA A100 SXM4 服務(wù)器
SXM 是 NVIDIA 專有的外形尺寸，可提高電力輸送、冷卻和帶寬。僅提供 PCIe 的云提供商通常不會(huì)指定他們的 H100 是 SXM 還是 PCIe。這是因?yàn)?PCIe 的性能要低得多，最多低 25%！這是因?yàn)?H100 PCIe 的內(nèi)存帶寬僅為 2 TB/s，而 SXM5 的內(nèi)存帶寬為 3.35 TB/s?？傮w而言，SXM 將 GPU 緊密地集成在服務(wù)器中，因此它們可以有效地作為單個(gè)超大 GPU 運(yùn)行。對于 A100 和 V100，性能上沒有太大差別，所以只要選擇更便宜的就行。
