def METHOD_NAME(framework_version, ecr_image, sagemaker_regions):
    with timeout(minutes=DEFAULT_TIMEOUT):
        estimator_parameter = {
            "entry_point": "main.py",
            "role": "SageMakerRole",
            "source_dir": resnet18_path,
            "instance_count": 1,
            "instance_type": MULTI_GPU_INSTANCE,
            "framework_version": framework_version,
        }
        job_name = utils.unique_name_from_base("test-pytorch-s3-plugin-gpu")
        invoke_pytorch_estimator(
            ecr_image, sagemaker_regions, estimator_parameter, job_name=job_name
        )