Tutorial: creare una definizione di EKS GPU lavoro Amazon

Al momento nvidia.com/gpu è supportato solo è supportato e il valore della risorsa impostato deve essere un numero intero. Non è possibile utilizzare frazioni di. GPU Per ulteriori informazioni, consulta Pianificazione GPUs nella Kubernetes documentazione.

Per registrare una definizione di GPU processo per AmazonEKS, esegui i seguenti comandi.

$ 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