메인 콘텐츠로 건너뛰기

Documentation Index

Fetch the complete documentation index at: https://wb-21fd5541-docs-2661.mintlify.app/llms.txt

Use this file to discover all available pages before exploring further.

W&B Launch로 GPU에서 TensorFlow 작업을 실행하려면 컨테이너 빌드에 맞춤형 베이스 이미지를 지정하세요. 이렇게 하면 생성된 컨테이너가 GPU에 액세스할 수 있습니다. 예를 들어, 리소스 설정의 builder.accelerator.base_image 키 아래에 이미지 태그를 추가합니다. 
{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}
W&B 0.15.6 이전 버전에서는 base_image의 부모 키로 accelerator 대신 cuda를 사용하세요.