オンデマンドキャパシティ予約 (ODCR) を使用してインスタンスを起動する

リソースグループを作成し、容量をグループ化します。

$ aws resource-groups create-group --name EC2CRGroup \ 
    --configuration '{"Type":"AWS::EC2::CapacityReservationPool"}' '{"Type":"AWS::ResourceGroups::Generic", "Parameters": [{"Name": "allowed-resource-types", "Values": ["AWS::EC2::CapacityReservation"]}]}'

#!/bin/bash
set -e

# Override run_instance attributes
cat > /opt/slurm/etc/pcluster/run_instances_overrides.json << EOF
{
    "my-queue": {
        "my-compute-resource": {
            "CapacityReservationSpecification": {
                "CapacityReservationTarget": {
                    "CapacityReservationId": "cr-abcdef01234567890"
                }
            }
        }
    }
}
EOF

リソースグループにキャパシティ予約を追加します。新しい ODCR を作成するたびに、それをグループ予約に追加します。ACCOUNT_ID はアカウント ID に、PLACEHOLDER_CAPACITY_RESERVATION はキャパシティ予約 ID に、REGION_ID は AWS リージョン ID (us-east-1 など) に置き換えてください。

$ aws resource-groups group-resources --region REGION_ID --group EC2CRGroup \
    --resource-arns arn:aws:ec2:REGION_ID:ACCOUNT_ID:capacity-reservation/PLACEHOLDER_CAPACITY_RESERVATION

ローカルコンピュータでポリシードキュメントを作成します。ACCOUNT_ID をアカウント ID に、REGION_ID を AWS リージョン ID (us-west-2 など) に置き換えます。

cat > policy.json << EOF
{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "RunInstancesInCapacityReservation",
            "Effect": "Allow",
            "Action": "ec2:RunInstances",
            "Resource": [
                "arn:aws:ec2:REGION_ID:ACCOUNT_ID:capacity-reservation/*",
                "arn:aws:resource-groups:REGION_ID:ACCOUNT_ID:group/*"
            ]
        }
    ]
}
EOF

作成した JSON ファイルを使用して、AWS アカウントに IAM ポリシーを作成します。

$ aws iam create-policy --policy-name RunInstancesCapacityReservation --policy-document file://policy.json

次のポストインストールスクリプトをインスタンス上でローカルに作成し、postinstall.sh という名前を付けます。

ACCOUNT_ID を AWS アカウント ID に、REGION_ID を AWS リージョン ID (us-east-1 など) に置き換えます。

#!/bin/bash
set -e

# Override run_instance attributes
cat > /opt/slurm/etc/pcluster/run_instances_overrides.json << EOF
{
    "my-queue": {
        "my-compute-resource": {
            "CapacityReservationSpecification": {
                "CapacityReservationTarget": {
                    "CapacityReservationResourceGroupArn": "arn:aws:resource-groups:REGION_ID:ACCOUNT_ID:group/EC2CRGroup"
                }
            }
        }
    }
}
EOF

Amazon S3 バケットにファイルをアップロードします。amzn-s3-demo-bucket は、特定の S3 バケット名に置き換えます。

$ aws s3 mb s3://amzn-s3-demo-bucket
aws s3 cp postinstall.sh s3://amzn-s3-demo-bucket/postinstall.sh

プレースホルダーを独自の値に置き換えて、ローカルクラスター設定を作成します。

Region: REGION_ID
Image:
  Os: alinux2
HeadNode:
  InstanceType: c5.2xlarge
  Ssh:
    KeyName: YOUR_SSH_KEY
  Iam:
    S3Access:
      - BucketName: amzn-s3-demo-bucket
    AdditionalIamPolicies:
      - Policy: arn:aws:iam::ACCOUNT_ID:policy/RunInstancesCapacityReservation
  ## This post-install script is executed after the node is configured.
  ## It is used to install scripts at boot time and specific configurations
  ## In the script below we are overriding the calls to RunInstance to force
  ## the provisioning of our my-queue partition to go through
  ## the On-Demand Capacity Reservation
  CustomActions:
    OnNodeConfigured:
      Script: s3://amzn-s3-demo-bucket/postinstall.sh
  Networking:
    SubnetId: YOUR_PUBLIC_SUBNET_IN_TARGET_AZ

Scheduling:
  Scheduler: slurm
  SlurmQueues:
    - Name: my-queue
      ComputeResources:
        - MinCount: 0
          MaxCount: 100
          InstanceType: p4d.24xlarge
          Name: my-compute-resource
          Efa:
            Enabled: true
      Networking:
        ## PlacementGroup:
        ##   Enabled: true ## Keep PG disabled if using targeted ODCR
        SubnetIds:
          - YOUR_PRIVATE_SUBNET_IN_TARGET_AZ

クラスターを作成します。

次のコマンドを使用してクラスターを作成します。cluster-config.yaml を構成ファイル名に、cluster-dl をクラスター名に、REGION_ID をリージョン ID (us-east-1 など) に置き換えます。

$ pcluster create-cluster --cluster-configuration cluster-config.yaml --cluster-name cluster-dl --region REGION_ID

クラスターが作成されると、ポストインストールスクリプトはヘッドノードで実行されます。このスクリプトは run_instances_overrides.json ファイルを作成し、RunInstances の呼び出しをオーバーライドして、パーティションのプロビジョニングがオンデマンドキャパシティ予約を経由するように強制します。

クラスタースケーリングを担当する AWS ParallelCluster デーモンは、起動される新しいインスタンスに自動的にこの設定を使用します。指定したパラメータがインスタンスのプロビジョニングに使用されていることを確認するには、以下のログファイルを確認します。

パラメータが正しければ、ログエントリには以下が含まれます。

Found RunInstances parameters override. Launching instances with: <parameters_list>