Configura la tua istanza di lavoro ibrida

A seconda del tuo 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 job ibrido utilizzando il seguente argomento di importazione e configurazione.

from braket.jobs.config import InstanceConfig

job = AwsQuantumJob.create(
    ...
    instance_config=InstanceConfig(instanceType="ml.g4dn.xlarge"), # Use NVIDIA T4 instance with 4 GPUs.
    ...
    ),

Se state eseguendo una simulazione incorporata e avete specificato un dispositivo locale nella configurazione del dispositivo, potete inoltre richiedere più di un'istanza InstanceConfig specificando instanceCount e impostandola su un valore 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.g4dn.xlarge", instanceCount=3), # Use 3 NVIDIA T4 instances
    ...
    ),

Quando utilizzi più istanze, valuta la possibilità di distribuire il tuo job ibrido utilizzando la funzionalità di parallelo dei dati. Vedi il seguente taccuino di esempio per maggiori dettagli su come vedere questo esempio di formazione Parallelize per QML.

Le tre tabelle seguenti elencano i tipi e le specifiche di istanza 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. Ma puoi regolare lo spazio di archiviazione nello stesso modo in cui configuriinstanceType. 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.g4dn.xlarge",
        volumeSizeInGb=50,
    ),
    ...
    ),

Configura il bucket predefinito in AwsSession

L'utilizzo della tua AwsSession istanza offre una maggiore flessibilità, ad esempio 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
)