在您自己的容器中運行 Braket 混合作業 - Amazon Braket

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

在您自己的容器中運行 Braket 混合作業

要使用自己的容器創建混合作業,請AwsQuantumJob.create()使用image_uri指定的參數調用。您可以使用隨選模擬器,或在 Braket 混合作業提供的傳統處理器上本機執行程式碼。QPU我們建議在真實運行之前在模擬器上測試您的代碼 SV1DM1,例如,或者在運行TN1之前QPU。

若要在傳統的處理器上執行程式碼,請透過更新來指定instanceCount您使用的InstanceConfiginstanceType請注意,如果您指定 instance_count > 1,則需要確保您的代碼可以在多個主機上運行。您可以選擇的執行個體數目上限為 5。例如:

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", # ...)
注意

使用裝置ARN追蹤您用作混合工作中繼資料的模擬器。可接受的值必須遵循格式device = "local:<provider>/<simulator_name>"。請記住,<provider>並且<simulator_name>必須僅由字母,數字,_-,和組成.。字串的長度限制為 256 個字元。

如果您打算使用,但BYOC並未使用 Braket SDK 來建立量子任務,則應將環境變AMZN_BRAKET_JOB_TOKEN數的值傳遞給CreateQuantumTask要求中的jobToken參數。如果不這樣做,量子任務將不會獲得優先級,並作為常規的獨立量子任務計費。