Executando trabalhos híbridos do Braket em seu próprio contêiner - Amazon Braket

Executando trabalhos híbridos do Braket em seu próprio contêiner

Para criar um trabalho híbrido com seu próprio contêiner, chame AwsQuantumJob.create() com o argumento image_uri especificado. Você pode usar uma QPU, um simulador sob demanda ou executar seu código localmente no processador clássico disponível com o Braket Hybrid Jobs. Recomendamos testar seu código em um simulador como SV1, DM1 ou TN1 antes de executar em uma QPU real.

Para executar seu código no processador clássico, especifique o instanceType e o instanceCount que você usa atualizando o InstanceConfig. Observe que, se você especificar um instance_count > 1, precisará garantir que seu código possa ser executado em vários hosts. O limite superior para o número de instâncias que você pode escolher é 5. Por exemplo:

job = AwsQuantumJob.create(
    source_module="source_dir",
    entry_point="source_dir.algorithm_script:start_here",
    image_uri="111122223333.dkr.ecr.us-west-2.amazonaws.com/my-byoc-container:latest",
    instance_config=InstanceConfig(instanceType="ml.p3.8xlarge", instanceCount=3),
    device="local:braket/braket.local.qubit",
    # ...)
nota

Use o ARN do dispositivo para rastrear o simulador que você usou como metadados do trabalho híbrido. Os valores aceitáveis devem seguir o formato device = "local:<provider>/<simulator_name>". Lembre-se que <provider> e <simulator_name> deve consistir apenas em letras, números, _, -, e .. O tamanho máximo da string é 256 caracteres.

Se você planeja usar o BYOC e não está usando o SDK do Braket para criar tarefas quânticas, deve passar o valor da variável ambiental AMZN_BRAKET_JOB_TOKEN para o parâmetro jobToken na solicitação CreateQuantumTask. Caso contrário, as tarefas quânticas não têm prioridade e são cobradas como tarefas quânticas autônomas regulares.