ドキュメント / 日本語ソースを見る ↗

インストールとクイックスタート

English | 简体中文 | 日本語 | Deutsch | Русский

ドキュメント · 次へ: プログラミング ガイド · 中文

1. エントリーポイントを選択してください

タスク エントリポイント
GPU カーネルを書き込みます ruda-kernel::dsl とデバイス ランタイム
行列の乗算、FFT、またはリダクションを使用します。 計算ライブラリ
テンソルとフレームワークの操作 テンソルとフレームワーク
モデルのトレーニングと状態の保存 トレーニングガイド
モデルをロードしてテキストを生成するか、画像を処理します モデル推論ガイド
デバイス バックエンドを統合する ドライバー API

ソース ワークスペースの Ruda を使用します。

2. NVIDIA 環境を準備する

Rust/Cargo、プラットフォーム用のリンカー ツールチェーン、NVIDIA GPU ドライバー、および CUDA ツールキットが必要です。 CUDA バックエンドには、NVRTC とツールキットのビルド依存関係が含まれています。直接 PTX を有効にしても、それらは削除されません。

ソース ルートから環境を確認します。

rustc --version --verbose
cargo --version
nvidia-smi
nvcc --version
cargo metadata --no-deps --format-version 1 --offline --locked

これらのコマンドは、Ruda をコンパイルしません。 --offline では、解決に必要な依存関係をローカルにキャッシュする必要があります。

CUDA_PATH を設定して、CUDA ツールキット ルートを選択します。 Windows では、複数のバージョンを含む親ではなく、インストールされているバージョンのディレクトリを指します。 CUDA インストール パス インターフェイス を参照してください。

3. サンプルをビルドする

ビルド環境の準備ができたら、次を実行します。

cargo build --locked -p ruda-driver-cuda --features direct-ptx --example ptx-runtime

ptx-runtime の例には、direct-ptx が必要です。この機能だけを有効にしても、デフォルトのコンパイラは変更されません。

4. コンパイル パスを選択して実行します。

別の PowerShell セッションでパスを 1 つ選択してください。続行する前に、コマンドの失敗を解決してください。

デフォルトの CUDA C++/NVRTC パス:

$env:RUDA_CUDA_COMPILER = 'nvrtc'
cargo run --locked -p ruda-driver-cuda --features direct-ptx --example ptx-runtime

直接 PTX パス:

$env:RUDA_CUDA_COMPILER = 'ptx'
$env:RUDA_PTX_VERSION = '8.0'
cargo run --locked -p ruda-driver-cuda --features direct-ptx --example ptx-runtime

PTX のバージョンは、ターゲット GPU およびドライバーと一致する必要があります。 PTX バックエンド リファレンス を参照してください。

この例では、さまざまな長さで FP32 加算を実行し、繰り返し実行される結果とテール センチネルをチェックし、キャッシュ カウンターを出力します。オプションのケースについては、例とチュートリアル を参照してください。

5. 開発を続ける

例のデバイスの選択、データのアップロード、およびカーネルの起動に従って、プログラミング ガイド を読んでください。ビルド、ドライバーの読み込み、または実行エラーの場合は、デバッグと診断 を使用して、失敗したステージを特定します。