Configura l'istanza di lavoro ibrida per eseguire lo script

A seconda dell'algoritmo, potresti avere requisiti diversi. Per impostazione predefinita, Amazon Braket esegue lo script dell'algoritmo su un'ml.m5.largeistanza. Tuttavia, puoi personalizzare questo tipo di istanza quando crei un lavoro ibrido utilizzando il seguente argomento di importazione e configurazione.

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 state eseguendo una simulazione incorporata e avete specificato un dispositivo locale nella configurazione del dispositivo, potrete inoltre richiedere più di un'istanza InstanceConfig specificando instanceCount e impostando che sia maggiore di una. Il limite massimo è 5. Ad esempio, puoi scegliere 3 istanze come segue.

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

Quando utilizzi più istanze, valuta la possibilità di distribuire il job ibrido utilizzando la funzionalità data parallel. Consulta il seguente taccuino di esempio per maggiori dettagli su come vedere questo esempio di formazione su Parallelize per QML.

Le tre tabelle seguenti elencano i tipi di istanze e le specifiche disponibili per le istanze standard, ad alte prestazioni e con accelerazione GPU.

Ogni istanza utilizza una configurazione predefinita di archiviazione dati (SSD) di 30 GB. È tuttavia possibile regolare lo spazio di archiviazione nello stesso modo in cui si configura ilinstanceType. L'esempio seguente mostra come aumentare lo spazio di archiviazione totale a 50 GB.

from braket.jobs.config import InstanceConfig

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

Configura il bucket predefinito in AwsSession

L'utilizzo della tua AwsSession istanza ti offre una maggiore flessibilità, come la possibilità di specificare una posizione personalizzata per il tuo bucket Amazon S3 predefinito. Per impostazione predefinita, an AwsSession ha una posizione del bucket Amazon S3 preconfigurata di. "amazon-braket-{id}-{region}" Tuttavia, hai la possibilità di sovrascrivere la posizione predefinita del bucket Amazon S3 durante la creazione di un. AwsSession Gli utenti possono facoltativamente passare un AwsSession oggetto al AwsQuantumJob.create() metodo, fornendo il aws_session parametro come illustrato nel seguente esempio di codice.

aws_session = AwsSession(default_bucket="amazon-braket-s3-demo-bucket")

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