자습서: Amazon EKS GPU 작업 정의 생성

현재는 nvidia.com/gpu만 지원되며, 설정하는 리소스 값은 정수여야 합니다. 의 일부를 사용할 수 없습니다GPU. 자세한 내용은의 일정을 GPUs 참조하세요. Kubernetes 설명서.

Amazon에 대한 GPU 작업 정의를 등록하려면 다음 명령을 EKS실행합니다.

$ cat <<EOF > ./batch-eks-gpu-jd.json
{
    "jobDefinitionName": "MyGPUJobOnEks_Smi",
    "type": "container",
    "eksProperties": {
        "podProperties": {
            "hostNetwork": true,
            "containers": [
                {
                    "image": "nvcr.io/nvidia/cuda:10.2-runtime-centos7",
                    "command": ["nvidia-smi"],
                    "resources": {
                        "limits": {
                            "cpu": "1",
                            "memory": "1024Mi",
                            "nvidia.com/gpu": "1"
                        }
                    }
                }
            ]
        }
    }
}
EOF

$ aws batch register-job-definition --cli-input-json file://./batch-eks-gpu-jd.json