AMD ROCm™ ソフトウェア

AMD ROCm™ は、サポートされている AMD GPU 上で AI およびハイパフォーマンス コンピューティング アプリケーションを開発するためのオープン ソフトウェア スタックです。AI モデルのトレーニングと提供、GPU アクセラレーション ソフトウェアの構築、GPU とノード全体での科学的ワークロードの実行に使用できます。

ROCm と CUDA は、別個の GPU コンピューティング プラットフォームです。ROCm は、サポートされている AMD GPU 向けのオープン ソフトウェア スタックを提供し、その HIP プログラミング モデルは CUDA ソース コードの移植パスを提供します。ツールと API は直接の互換性がないため、移行作業はアプリケーションとその依存関係によって異なります。

常にそうとは限りません。サポートされている AI フレームワークに基づいて構築されたアプリケーションは、ROCm 対応バージョンのフレームワークから開始できる場合があります。CUDA API を呼び出すコード、または CUDA 固有のライブラリに依存するコードには変更が必要な場合があります。HIPIFY は、サポートされている CUDA コードを HIP に変換するのに役立ちます。その後、開発者は正確性をテストし、必要に応じてパフォーマンスを調整する必要があります。

ROCm は、PyTorch、TensorFlow、JAX を使用した AI 開発と、vLLM、SGLang を使用したモデル サービングに対応しています。サポートされるバージョンは、ROCm リリースとシステム構成によって異なります。フレームワークのバージョンを選択する前に、最新の互換性マトリックスを確認してください。

まず、GPU とオペレーティング システムの互換性マトリクスを確認します。その後、AMD のインストール ガイドに従って設定します。AMD Infinity ハブは事前に設定されたコンテナーを提供し、ROCm CLI はサポートされているシステムのセットアップと診断を支援します。拡張する前に、サンプル ワークロードを実行して環境を確認してください。