使用方法
説明
docker run をラップし、W&B 環境変数を自動的に設定します。
すべての引数を docker run に渡すとともに、次の値を設定します。
-
WANDB_API_KEY: ログインしている場合、コンテナが W&B に対して認証できるよう、現在の API キーを設定します。
- WANDB_DOCKER: 引数からイメージを検出できる場合、解決されたイメージ ID を設定します。これにより、W&B は実行に使用された Docker イメージを追跡できます。
nvidia-docker が検出され、--runtime がまだ設定されていない場合は、--runtime nvidia を自動的に追加します。
たとえば、「my-image」コンテナ内で Python スクリプト train.py を実行するには、次のようにします。