Ejecución de trabajos híbridos de Braket en su propio contenedor - Amazon Braket

Ejecución de trabajos híbridos de Braket en su propio contenedor

Para crear un trabajo híbrido con su propio contenedor, llame a AwsQuantumJob.create() con el argumento de image_uri especificado. Puede utilizar una QPU, un simulador bajo demanda o ejecutar su código localmente en el procesador clásico disponible con los trabajos híbridos de Braket. Recomendamos probar el código en un simulador como SV1, DM1 o TN1 antes de ejecutarlo en una QPU real.

Para ejecutar el código en el procesador clásico, especifique el instanceType y el instanceCount que utiliza actualizando la InstanceConfig. Tenga en cuenta que si especifica un instance_count > 1, debe asegurarse de que el código puede ejecutarse en varios hosts. El límite máximo de instancias que puede elegir es de 5. Por ejemplo:

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

Utilice el ARN del dispositivo para realizar un seguimiento del simulador que utilizó como metadatos del trabajo híbrido. Los valores aceptables deben seguir el formato device = "local:<provider>/<simulator_name>". Recuerde que <provider> y <simulator_name> debe constar únicamente de letras, números, _, - y . El tamaño de la cadena se limita a 256 caracteres.

Si planea BYOC y no usa el SDK de Braket para crear tareas cuánticas, debe pasar el valor de la variable de entorno AMZN_BRAKET_JOB_TOKEN al parámetro jobToken en la solicitud CreateQuantumTask. De lo contrario, las tareas cuánticas no tienen prioridad y se consideran tareas cuánticas normales e independientes.