特斯拉推出搭載NVIDIA A100 GPU頂尖自動駕駛汽車訓練超級計算機

“不可思議”的GPU集群賦能Autopilot系統(tǒng)及全自動駕駛的AI開發(fā)工作。

為了應對這項有史以來最大的計算挑戰(zhàn)之一，特斯拉需要配備前所未有的算力。

在本周的CVPR（國際計算機視覺與模式識別會議）上，汽車制造商特斯拉的AI高級總監(jiān)Andrej Karpathy公布了公司內部用于訓練Autopilot與自動駕駛深度神經(jīng)網(wǎng)絡的超級計算機。這個集群使用了720個節(jié)點的8個NVIDIA A100 Tensor Core GPU（共5760個GPU），實現(xiàn)了1.8 exaflops級別的超強性能。

Karpathy表示：“這是一臺性能卓越的超級計算機，以FLOPS計，它大概可以在全球超級計算機中排到第5位”。

通過將汽車行業(yè)前所未有的算力作為其研發(fā)周期的核心，特斯拉讓其自動駕駛汽車工程師能夠使用前沿技術來高效完成工作。

NVIDIA A100 GPU為全球最強的數(shù)據(jù)中心提供各種尺度的加速。A100 GPU基于NVIDIA Ampere架構打造，其性能比上一代產(chǎn)品高出20倍，并且可以劃分成7個GPU實例，動態(tài)地適配不同的需求。

使用100多萬輛在路上行駛的Tesla汽車（的數(shù)據(jù)）來持續(xù)優(yōu)化和迭代新功能，是Tesla自動駕駛的垂直整合之道，而GPU集群在其中扮演了重要角色。

從汽車到數(shù)據(jù)中心

特斯拉的數(shù)據(jù)循環(huán)始于汽車。“影子模式”在不實際控制車輛的情況下，悄無聲息地執(zhí)行著感知和預測深度神經(jīng)網(wǎng)絡（DNN）。

任何錯誤預測、誤識別都會被記錄下來。隨后，這些實例會被特斯拉工程師所用，來創(chuàng)建/擴充一個包含多種復雜場景的訓練數(shù)據(jù)集，以完善DNN。

當前已經(jīng)收集了100萬個以每秒36幀的速度記錄的10秒片段，總數(shù)據(jù)量高達1.5PB。在數(shù)據(jù)中心，DNN基于這些場景被反復訓練直到性能可接受。最后，DNN被發(fā)送回車內，并開始下一輪（數(shù)據(jù)）循環(huán)。

Karpathy表示，以這種方式在如此大量的數(shù)據(jù)上訓練DNN需要龐大的計算能力，特斯拉為此建立并部署了內置高性能A100 GPU的最新一代超級計算機。

連續(xù)迭代

除了全方位的訓練之外，特斯拉的超級計算機還為自動駕駛汽車工程師提供了在開發(fā)過程中進行實驗和迭代所需的性能。

Karpathy表示，特斯拉目前部署的DNN結構可以讓一個由20名工程師組成的團隊同時在一個網(wǎng)絡上工作，通過隔離不同功能來實現(xiàn)并行開發(fā)。

這些DNN運行訓練數(shù)據(jù)集的速度會比之前快速迭代時還更快。

Karpathy表示：“計算機視覺是我們一切工作的基礎，并且也是實現(xiàn)Autopilot的關鍵。為此，我們必須訓練一個龐大的神經(jīng)網(wǎng)絡并進行大量實驗。這也是我們在算力方面投入大量資金的原因?！?/p>

歡迎自動駕駛領域的各位有志之士加入NVIDIA Developer Program，復制鏈接“https://developer.nvidia.com/login”在瀏覽器中打開即可注冊（請在Industry Segment注冊選項中選擇Automotive）。

編輯：jq