로컬 실행을 SDK v2로 업그레이드

로컬 실행은 V1과 V2 모두에서 비슷합니다. 두 버전에서 컴퓨팅 대상을 설정할 때 "로컬" 문자열을 사용합니다.

이 문서에서는 SDK v1과 SDK v2의 시나리오를 비교합니다.

로컬 실행 제출

• SDK v1

  from azureml.core import Workspace, Experiment, Environment, ScriptRunConfig
  
  # connect to the workspace
  ws = Workspace.from_config()
  
  # define and configure the experiment
  experiment = Experiment(workspace=ws, name='day1-experiment-train')
  config = ScriptRunConfig(source_directory='./src',
                              script='train.py',
                              compute_target='local')
  
  # set up pytorch environment
  env = Environment.from_conda_specification(
      name='pytorch-env',
      file_path='pytorch-env.yml')
  config.run_config.environment = env
  
  run = experiment.submit(config)
  
  aml_url = run.get_portal_url()
  print(aml_url)
  

• SDK v2

  #import required libraries
  from azure.ai.ml import MLClient, command
  from azure.ai.ml.entities import Environment
  from azure.identity import DefaultAzureCredential
  
  #connect to the workspace
  ml_client = MLClient.from_config(DefaultAzureCredential())
  
  # set up pytorch environment
  env = Environment(
      image='mcr.microsoft.com/azureml/openmpi3.1.2-ubuntu18.04',
      conda_file='pytorch-env.yml',
      name='pytorch-env'
  )
  
  # define the command
  command_job = command(
      code='./src',
      command='train.py',
      environment=env,
      compute='local',
  )
  
  returned_job = ml_client.jobs.create_or_update(command_job)
  returned_job
  

SDK v1 및 SDK v2의 주요 기능 매핑

SDK v1의 기능	SDK v2의 대략적인 매핑
experiment.submit	MLCLient.jobs.create_or_update

다음 단계

• Azure Machine Learning을 사용하여 모델 학습