builder.accelerator.base_image 키 아래에 이미지 태그를 추가합니다.
base_image의 부모 키로 accelerator 대신 cuda를 사용하세요.Help us improve these docs. Take our quick survey.
builder.accelerator.base_image 키 아래에 이미지 태그를 추가합니다.
{
"gpus": "all",
"builder": {
"accelerator": {
"base_image": "tensorflow/tensorflow:latest-gpu"
}
}
}
base_image의 부모 키로 accelerator 대신 cuda를 사용하세요.