使用方法
Description
W&B が設定された Docker コンテナーでコードを実行します。 Docker コンテナーを起動し、WANDB_DOCKERS 環境変数を注入して、現在の作業ディレクトリをデフォルトで/app にマウントします。ログインしている場合は WANDB_API_KEY を注入します。wandb がインストールされていることを確実にするため、コンテナーのエントリポイントを上書きします。
追加の引数を渡すと、イメージ名の前に docker run へ挿入されます。指定がない場合はデフォルトのイメージを使用します。
--jupyter を使用すると、JupyterLab をインストールしてポート 8888 で起動します。ホストで NVIDIA Docker が利用可能な場合は、NVIDIA ランタイムを自動的に有効にします。
Docker がホストマシンにインストールされ、実行されている必要があります。
既存の docker run コマンドに W&B 環境変数を注入し、エントリポイントを変更せずに使用するには、wandb docker-run を使用してください。
たとえば、デフォルトのイメージを実行し、データセットをコンテナーにマウントするには: