SageMaker Profiler のトラブルシューティング

次の質問と回答のペアを使用して、SageMaker Profiler 使用中の問題のトラブルシューティングを行います。

Q. エラーメッセージ ModuleNotFoundError: No module named 'smppy' が表示されます

2023 年 12 月以降、SageMaker Profiler Python パッケージの名前が smppy から smprof に変更され、パッケージ名の重複が解決されました。smppy はオープンソースパッケージで既に使用されています。

したがって、2023 年 12 月以前から smppy を使用していて、この ModuleNotFoundError 問題が発生した場合は、最新の smprof パッケージがインストールされているか、最新の SageMaker Profiler がプリインストールされた SageMaker AI フレームワークイメージ のいずれかを使用しているときに、トレーニングスクリプトに古いパッケージ名が含まれていることが原因である可能性があります。この場合、トレーニングスクリプト全体で、smppy と言及されている箇所をすべて smprof に置き換えてください。

トレーニングスクリプトで SageMaker Profiler Python パッケージ名を更新する際は、使用するパッケージ名のバージョンに関して混乱が起きないよう、次のコードスニペットに示すように、条件付きインポートステートメントの使用を検討してください。

try:
    import smprof 
except ImportError:
    # backward-compatability for TF 2.11 and PT 1.13.1 images
    import smppy as smprof

また、最新の PyTorch または TensorFlow バージョンへのアップグレード中に smppy を使用している場合は、(オプション) SageMaker Profiler Python パッケージをインストールする の指示に従って最新の smprof パッケージをインストールしてください。

Q. エラーメッセージ ModuleNotFoundError: No module named 'smprof' が表示されます

まず、公式にサポートされている SageMaker AI Framework Containers のいずれかを使用していることを確認してください。これらのいずれかを使用しない場合は、「(オプション) SageMaker Profiler Python パッケージをインストールする」の指示に従って smprof パッケージをインストールできます。

Q. ProfilerConfig をインポートできません

SageMaker Python SDK を使用してジョブランチャースクリプトに ProfilerConfig をインポートできない場合、ローカル環境または Jupyter カーネルに SageMaker Python SDK のかなり古いバージョンがある可能性があります。SDK を最新バージョンにアップグレードしていることを確認します。

$ pip install --upgrade sagemaker

Q. エラーメッセージ aborted: core dumped when importing smprof into my training script が表示されます

以前のバージョンの smprof では、この問題は PyTorch 2.0+ と PyTorch Lightning で発生します。この問題を解決するには、「(オプション) SageMaker Profiler Python パッケージをインストールする」の指示に従って最新の smprof パッケージをインストールします。

Q. SageMaker Studio で SageMaker Profiler UI が見つかりません。どうすれば見つかりますか?

SageMaker AI コンソールにアクセスできる場合は、次のいずれかのオプションを選択します。

ドメインユーザーで SageMaker AI コンソールにアクセスできない場合は、SageMaker Studio Classic からアプリケーションにアクセスできます。この場合、以下のオプションを選択します。