earth2studio-install

Earth2Studio インストールスキル

パッケージを自動でインストールしない

ユーザーに代わって、パッケージをインストール、アップグレード、変更してはいけません。正確なコマンドを提示し、ユーザー自身に実行してもらいます。例外はありません。

禁止事項： pip install、uv pip install、uv add、 uv sync、conda install、apt install、またはあらゆるパッケージマネージャーの実行。

代わりに： 正確なコマンドを提示し、ユーザーに実行を依頼します。 そのパッケージが必要な理由を説明します。

パッケージが必要な場合：

1. パッケージを特定する
2. 正確なコマンドを提示する
3. 必要な理由を説明する
4. ユーザーが実行したことを確認するまで待つ

ユーザーが「そのままインストールして」と言った場合も、コマンドを提示し、ユーザー自身に実行してもらいます。

目的

ユーザーが用途に応じて、Earth2Studio とオプションのモデル依存関係を正しくインストールできるよう支援します。このスキルでは、パッケージのインストール、オプションの選択、環境変数の設定、インストールの検証を扱います。

前提条件

• Python 3.10 以降（3.13 を推奨）
• GPU 用の追加機能を使う場合は、互換性のあるドライバーを備えた CUDA 対応 GPU
• uv（推奨）または pip パッケージマネージャー
• インターネット接続（パッケージは PyPI と GitHub からインストールされます）

ユーザーによる Earth2Studio とオプションのモデル依存関係のインストールを支援します。あなたの役割は、ユーザーの用途に応じてパッケージを正しくインストールすることだけです。推論コードを書いたり、ワークフローを構成したりしないでください。

基本原則：ドキュメントを正とする

Earth2Studio のインストールコマンド、バージョンタグ、追加機能の名前はリリースごとに変わります。インストールコマンドを実行または推奨する前に、必ず最新のインストールドキュメントを取得してください：

https://nvidia.github.io/earth2studio/userguide/about/install.html

ページを確認し、現在のバージョンタグ、利用可能な追加機能、特別なビルドに関する注意事項を把握します。以下の手順は構成上のガイドです。具体的なコマンドは最新のページを参照してください。

手順

ステップ 1. 最新のドキュメントを取得する

上記のインストール URL に WebFetch を使用します。次の情報を抽出します：

• 現在のリリースバージョンタグ（例：@0.14.0）
• カテゴリ別の利用可能なオプション追加機能
• 既知のビルド上の注意点（例：pip での --no-build-isolation、事前の手動インストール）

以降の手順では、この情報を作業メモリに保持します。

ステップ 2. ユーザーの環境を把握する

質問は最大 3 つまでとし、ユーザーがすでに回答している内容は尋ねません：

1. パッケージマネージャー — uv（推奨）または pip のどちらですか？不明な場合は uv を推奨し、https://docs.astral.sh/uv/getting-started/installation/ を案内します
2. プロジェクトの状況 — 新規プロジェクトですか、それとも既存プロジェクトへの追加ですか？
3. Python のバージョン — ドキュメントで推奨されているバージョン（現在は 3.13）を推奨します

ステップ 3. 基本インストール

ユーザーの回答に基づいて、最新のドキュメントに記載されたコマンドを提示します：

• uv は、URL ベースの推移的依存関係に対応するため、Git ソースを使用します（PyPI は使用しません）
• pip は PyPI からインストールしますが、一部の追加機能では事前の手動インストールが必要です

ユーザーがインストールを実行した後、次の手順で検証します：

import earth2studio
earth2studio.__version__

ステップ 4. モデルと追加機能を選択する

利用可能な追加機能を用途別に提示します。ユーザーが何をする予定か尋ねてください。すべての選択肢を先回りして提示しないでください。ドキュメントに記載されたカテゴリは次のとおりです：

カテゴリ	追加機能の例
予報（予測）	aifs, aurora, graphcast, pangu, sfno, stormcast, ...
診断（後処理）	corrdiff, climatenet, precip-afno, ...
データ同化（ベータ）	da-healda, da-interp, da-stormcast
サブモジュール	data, perturbation, statistics

正確な一覧は最新のドキュメントを参照し、この表ではなくそちらを案内します。

質問：

1. どのモデルを使う予定ですか？
2. サブモジュールの追加機能（データソース、摂動手法、統計）が必要ですか？
3. それともすべてをインストールしますか？（uv のみ：--extra all）

ステップ 5. 選択した追加機能をインストールする

選択内容に応じて、ライブドキュメントに記載されている正確なコマンドを提示します。 伝えるべき主な注意事項：

• ビルドに時間がかかります：flash-attention（AIFS バリアント）、natten （Atlas、StormScope）、torch-harmonics の CUDA 拡張（FCN3、SFNO） — 10〜30 分以上かかる場合があります
• pip 固有の手動手順：モデルによっては、--no-build-isolation を指定するか、 earth2grid、torch-harmonics、makani などのパッケージを 事前にインストールする必要があります
• データ同化モデル：CuPy と cuDF（CUDA 12）が必要です

ステップ 6. 設定（強制ではなく提案）

必要に応じて（例：ディスク容量が限られている場合、共有ファイルシステムを使用する場合、 CI 環境の場合）、設定したい環境変数を伝えます。

変数	用途
EARTH2STUDIO_CACHE	一般的なキャッシュディレクトリ
EARTH2STUDIO_DATA_CACHE	データソースのキャッシュ（一般的なキャッシュを上書き）
EARTH2STUDIO_MODEL_CACHE	モデルチェックポイントのキャッシュ（一般的なキャッシュを上書き）
EARTH2STUDIO_PACKAGE_TIMEOUT	モデルのダウンロードにかける最大秒数

トラブルシューティング

インストールに失敗した場合は、以下を案内してください。

• https://nvidia.github.io/earth2studio/userguide/support/troubleshooting.html
• https://nvidia.github.io/earth2studio/userguide/support/faq.html

よくある問題：

• PyTorch/CUDA の不一致：まず torch.cuda.is_available() を確認します
• flash-attention のビルド失敗：CUDA ツールキットのバージョンが PyTorch の CUDA と一致している必要があります
• ONNX Runtime GPU：使用する CUDA に応じたバージョンのインストールが必要な場合があります
• ecCodes が見つからない：GRIB データの処理に必要です。sudo apt-get install libeccodes-dev（Debian/Ubuntu）または conda install -c conda-forge eccodes でインストールします
• Python.h: No such file or directory：Python 開発用ヘッダーが不足しています。 sudo apt-get install python3-dev でインストールします

制限事項

• 依存関係の不足とは無関係な実行時エラーには対応できません
• モデルチェックポイントのダウンロードは対象外です（初回の推論時に行われます）
• data 追加機能の範囲を超えるデータソースの設定は対象外です
• 推論コードやトレーニングコードの作成、および Earth2Studio のワークフローの構成はできません

対応範囲と対象外

対応範囲： パッケージのインストール、オプションの追加機能の選択、環境変数の設定、インストールの検証。

対象外： 推論コードやトレーニングコードの作成、Earth2Studio のワークフローの構成、 data 追加機能の範囲を超えるデータソースの設定、 モデルチェックポイントのダウンロード（実行時に行われます）、依存関係の不足とは無関係な 実行時エラーのトラブルシューティング。