GPU 를 사용하는 TensorFlow 작업의 경우, 컨테이너 빌드를 위한 커스텀 베이스 이미지를 지정하세요. 이를 통해 Runs 가 진행되는 동안 GPU 가 적절하게 활용되도록 보장할 수 있습니다. 리소스 설정의Documentation Index
Fetch the complete documentation index at: https://wb-21fd5541-update-training-api-26.mintlify.app/llms.txt
Use this file to discover all available pages before exploring further.
builder.accelerator.base_image 키 아래에 이미지 태그를 추가하세요. 예시는 다음과 같습니다:
base_image 의 부모 키로 accelerator 대신 cuda 를 사용하세요.