Vertex AI SDK for Python をインストールする

Vertex AI SDK for Python を使用して、ML ワークフローを自動化します。このトピックでは、Vertex AI SDK for Python のインストール方法について説明します。Vertex AI SDK の詳細については、次のリソースをご覧ください。

Vertex AI SDK for Python のインストールの手順は次のとおりです。

  1. 分離された Python 環境を作成する
  2. Vertex AI SDK パッケージをインストールする
  3. Vertex AI SDK を初期化する

分離された Python 環境を作成する

Python のベスト プラクティスとして、プロジェクトごとに分離された Python 環境に Vertex AI SDK をインストールすることをおすすめします。これにより、依存関係、バージョン、権限の競合を防ぐことができます。シェルでコマンドラインを使用する、またはノートブックを使用するための分離された環境を作成できます。

コマンドラインを使用するときに分離された環境を作成するには、venv 環境をアクティブにしますvenv 環境がアクティブになったら、Vertex AI SDK をインストールして Python スクリプトを実行できます。詳細については、venv を使用して依存関係を分離すると、Python 開発環境を設定するをご覧ください。

分離環境でノートブックを使用するには、ノートブック インスタンスを作成します。ノートブック インスタンスを作成したら、それを使用して Vertex AI SDK をインストールし、Python スクリプトを実行します。詳細については、ユーザー管理のノートブック インスタンスを作成するをご覧ください。

Vertex AI SDK パッケージをインストールまたは更新する

Vertex AI SDK をインストールまたは更新するには、仮想環境で次のコマンドを実行します。

pip install --upgrade google-cloud-aiplatform

Vertex AI SDK を初期化する

Vertex AI SDK for Python をインストールしたら、Vertex AI と Google Cloud の詳細を使用して SDK を初期化する必要があります。たとえば、SDK を初期化するときに、プロジェクト名、リージョン、ステージング済み Cloud Storage バケットなどの情報を指定します。Vertex AI SDK を初期化するメソッドの例を次に示しす。

def init_sample(
    project: Optional[str] = None,
    location: Optional[str] = None,
    experiment: Optional[str] = None,
    staging_bucket: Optional[str] = None,
    credentials: Optional[google.auth.credentials.Credentials] = None,
    encryption_spec_key_name: Optional[str] = None,
    service_account: Optional[str] = None,
):

    import vertexai

    vertexai.init(
        project=project,
        location=location,
        experiment=experiment,
        staging_bucket=staging_bucket,
        credentials=credentials,
        encryption_spec_key_name=encryption_spec_key_name,
        service_account=service_account,
    )

次のステップ