Configure a instância de trabalho híbrida para executar seu script

Dependendo do seu algoritmo, você pode ter requisitos diferentes. Por padrão, o Amazon Braket executa seu script de algoritmo em uma instância. ml.m5.large No entanto, você pode personalizar esse tipo de instância ao criar um trabalho híbrido usando o seguinte argumento de importação e configuração.

from braket.jobs.config import InstanceConfig

job = AwsQuantumJob.create(
    ...
    instance_config=InstanceConfig(instanceType="ml.p3.8xlarge"), # Use NVIDIA Tesla V100 instance with 4 GPUs.
    ...
    ),

Se você estiver executando uma simulação incorporada e tiver especificado um dispositivo local na configuração do dispositivo, poderá solicitar adicionalmente mais de uma instância no InstanceConfig especificando instanceCount e configurando-a como maior que uma. O limite superior é 5. Por exemplo, você pode escolher 3 instâncias da seguinte maneira.

from braket.jobs.config import InstanceConfig
job = AwsQuantumJob.create(
    ...
    instance_config=InstanceConfig(instanceType="ml.p3.8xlarge", instanceCount=3), # Use 3 NVIDIA Tesla V100
    ...
    ),

Ao usar várias instâncias, considere distribuir seu trabalho híbrido usando o recurso de data parallel. Consulte o exemplo de caderno a seguir para obter mais detalhes sobre como ver esse treinamento do Parallelize, por exemplo. QML

As três tabelas a seguir listam os tipos e especificações de instância disponíveis para instâncias de computação padrão, otimizadas para computação e aceleradas.

Cada instância usa uma configuração padrão de armazenamento de dados (SSD) de 30 GB. Mas você pode ajustar o armazenamento da mesma forma que configura instanceType o. O exemplo a seguir mostra como aumentar o armazenamento total para 50 GB.

from braket.jobs.config import InstanceConfig

job = AwsQuantumJob.create(
    ...
    instance_config=InstanceConfig(
        instanceType="ml.p3.8xlarge",
        volumeSizeInGb=50,
    ),
    ...
    ),

Configure o bucket padrão em AwsSession

A utilização de sua própria AwsSession instância oferece maior flexibilidade, como a capacidade de especificar um local personalizado para seu bucket padrão do Amazon S3. Por padrão, um AwsSession tem uma localização pré-configurada de bucket do Amazon S3 de. f"amazon-braket-{id}-{region}" No entanto, você tem a opção de substituir a localização padrão do bucket do Amazon S3 ao criar um. AwsSession Opcionalmente, os usuários podem passar um AwsSession objeto para o AwsQuantumJob.create() método, fornecendo o aws_session parâmetro conforme demonstrado no exemplo de código a seguir.

aws_session = AwsSession(default_bucket="amzn-s3-demo-bucket")

# then you can use that AwsSession when creating a hybrid job
job = AwsQuantumJob.create(
    ...
    aws_session=aws_session
)