Créez une instance de bloc-notes Braket à l'aide de AWS CloudFormation

Astuce

Découvrez les bases de l'informatique quantique avec AWS ! Inscrivez-vous au plan d'apprentissage numérique Amazon Braket et obtenez votre propre badge numérique après avoir suivi une série de cours de formation et une évaluation numérique.

Vous pouvez l'utiliser AWS CloudFormation pour gérer vos instances de bloc-notes Amazon Braket. Les instances de bloc-notes Braket sont basées sur Amazon SageMaker AI. Avec CloudFormation, vous pouvez approvisionner une instance de bloc-notes avec un fichier modèle qui décrit la configuration prévue. Le fichier modèle est écrit au YAML format JSON OR. Vous pouvez créer, mettre à jour et supprimer des instances de manière ordonnée et reproductible. Cela peut vous être utile lorsque vous gérez plusieurs instances de bloc-notes Braket chez vous Compte AWS.

Après avoir créé un CloudFormation modèle pour un bloc-notes Braket, vous pouvez l'utiliser AWS CloudFormation pour déployer la ressource. Pour plus d'informations, consultez la section Création d'une pile sur la AWS CloudFormation console dans le guide de AWS CloudFormation l'utilisateur.

Pour créer une instance de bloc-notes Braket à l'aide de CloudFormation, vous devez effectuer les trois étapes suivantes :

1. Créez un script de configuration du cycle de vie Amazon SageMaker AI.

2. Créez un rôle AWS Identity and Access Management (IAM) à assumer par l' SageMaker IA.

3. Créez une instance de bloc-notes SageMaker AI avec le préfixe amazon-braket-

Vous pouvez réutiliser la configuration du cycle de vie pour tous les blocs-notes Braket que vous créez. Vous pouvez également réutiliser le IAM rôle pour les blocs-notes Braket auxquels vous attribuez les mêmes autorisations d'exécution.

Étape 1 : créer un script de configuration du cycle de vie Amazon SageMaker AI

Utilisez le modèle suivant pour créer un script de configuration du cycle de vie de l'SageMaker IA. Le script personnalise une instance de bloc-notes SageMaker AI pour Braket. Pour les options de configuration de la CloudFormation ressource de cycle de vie, voir AWS: SageMaker : : : NotebookInstanceLifecycleConfig dans le guide de AWS CloudFormation l'utilisateur.

BraketNotebookInstanceLifecycleConfig:
    Type: "AWS::SageMaker::NotebookInstanceLifecycleConfig"
    Properties:
      NotebookInstanceLifecycleConfigName: BraketLifecycleConfig-${AWS::StackName}
      OnStart:
      - Content:
          Fn::Base64: |
            #!/usr/bin/env bash
            sudo -u ec2-user -i ≪EOS
            curl -o braket-notebook-lcc.zip https://d3ded4lzb1lnme.cloudfront.net/notebook/braket-notebook-lcc.zip
            unzip braket-notebook-lcc.zip
            ./install.sh
            EOS
            
            exit 0

Étape 2 : créer le IAM rôle assumé par Amazon SageMaker AI

Lorsque vous utilisez une instance de bloc-notes Braket, l' SageMaker IA effectue des opérations en votre nom. Supposons, par exemple, que vous utilisiez un bloc-notes Braket à l'aide d'un circuit sur un appareil compatible. Dans l'instance de bloc-notes, l' SageMaker IA exécute l'opération sur Braket pour vous. Le rôle d'exécution du bloc-notes définit les opérations exactes que l' SageMaker IA est autorisée à exécuter en votre nom. Pour plus d'informations, consultez SageMaker la section Rôles de l' SageMaker IA dans le guide du développeur Amazon AI.

Utilisez l'exemple suivant pour créer un rôle d'exécution de bloc-notes Braket avec les autorisations requises. Vous pouvez modifier les politiques en fonction de vos besoins.

Note

Assurez-vous que le rôle dispose d'une autorisation pour les s3:GetObject opérations s3:ListBucket et sur les compartiments Amazon S3 préfixées par. braketnotebookcdk-" Le script de configuration du cycle de vie nécessite ces autorisations pour copier le script d'installation du bloc-notes Braket.

ExecutionRole: 
    Type: "AWS::IAM::Role"
    Properties: 
      RoleName: !Sub AmazonBraketNotebookRole-${AWS::StackName}
      AssumeRolePolicyDocument: 
        Version: "2012-10-17"
        Statement: 
        - 
          Effect: "Allow"
          Principal: 
            Service: 
              - "sagemaker.amazonaws.com"
          Action: 
          - "sts:AssumeRole"
      Path: "/service-role/"
      ManagedPolicyArns:
        - arn:aws:iam::aws:policy/AmazonBraketFullAccess
      Policies: 
        - 
          PolicyName: "AmazonBraketNotebookPolicy"
          PolicyDocument: 
            Version: "2012-10-17"
            Statement: 
              - Effect: Allow
                Action:
                  - s3:GetObject
                  - s3:PutObject
                  - s3:ListBucket
                Resource:
                  - arn:aws:s3:::amazon-braket-*
                  - arn:aws:s3:::braketnotebookcdk-*
              - Effect: "Allow"
                Action: 
                  - "logs:CreateLogStream"
                  - "logs:PutLogEvents"
                  - "logs:CreateLogGroup"
                  - "logs:DescribeLogStreams"
                Resource:
                  - !Sub "arn:aws:logs:*:${AWS::AccountId}:log-group:/aws/sagemaker/*"
              - Effect: "Allow"
                Action: 
                  - braket:*
                Resource: "*"

Étape 3 : créer une instance de bloc-notes Amazon SageMaker AI avec le préfixe amazon-braket-

Utilisez le script de cycle de vie de l' SageMaker IA et le IAM rôle créé aux étapes 1 et 2 pour créer une instance de bloc-notes SageMaker AI. L'instance de bloc-notes est personnalisée pour Braket et est accessible via la console Amazon Braket. Pour plus d'informations sur les options de configuration de cette CloudFormation ressource, consultez AWS: : SageMaker : : NotebookInstance dans le guide de AWS CloudFormation l'utilisateur.

BraketNotebook:
    Type: AWS::SageMaker::NotebookInstance
    Properties: 
      InstanceType: ml.t3.medium
      NotebookInstanceName: !Sub amazon-braket-notebook-${AWS::StackName}
      RoleArn: !GetAtt ExecutionRole.Arn
      VolumeSizeInGB: 30
      LifecycleConfigName: !GetAtt BraketNotebookInstanceLifecycleConfig.NotebookInstanceLifecycleConfigName