특정 작업에 TPU 메모리가 할당되는 방식을 지정할 수 없습니다.
JAX 특정 TPU 성능 문제에 대한 자세한 내용은 JAX에서 TPU 사용 시 성능 참고사항을 참조하세요.
TPU 문제 해결
TPU가 실행 중인지 확인하려면 어떻게 해야 하나요?
세부정보
JAX가 'GPU/TPU를 찾을 수 없으며 CPU로 돌아갑니다.'를 출력하지 않는 한 모든 항목이 TPU에서 실행됩니다.
여러 TPU 기기가 표시되는 경우 jax.devices()를 확인하여 TPU가 활성 상태인지 확인하거나, assert jax.devices()[0].platform == 'tpu'를 사용하여 프로그래매틱 방식으로 확인할 수 있습니다.
RuntimeError: 백엔드 'tpu'를 초기화할 수 없음: UNAVAILABLE: 사용 가능한 TPU 플랫폼이 없습니다.
세부정보
이 런타임 오류 메시지 또는 TPU VM의 /tmp/tpu_logs/tpu_driver.WARNING에서 다음 로그를 확인한 경우:
W1118 17:40:20.985243 23901 tpu_version_flag.cc:57] No hardware is found. Using default TPU version:xxxxxx
이는 잘못된 TPU VM 버전에서 실행 중일 가능성을 나타냅니다.
[[["이해하기 쉬움","easyToUnderstand","thumb-up"],["문제가 해결됨","solvedMyProblem","thumb-up"],["기타","otherUp","thumb-up"]],[["이해하기 어려움","hardToUnderstand","thumb-down"],["잘못된 정보 또는 샘플 코드","incorrectInformationOrSampleCode","thumb-down"],["필요한 정보/샘플이 없음","missingTheInformationSamplesINeed","thumb-down"],["번역 문제","translationIssue","thumb-down"],["기타","otherDown","thumb-down"]],["최종 업데이트: 2025-09-04(UTC)"],[],[],null,["# Troubleshooting JAX - TPU\n=========================\n\nThis guide provides pointers to JAX troubleshooting information to\nhelp you identify and resolve problems you might encounter while training\nJAX models on Cloud TPU.\n\nFor a more general guide to\ngetting started with Cloud TPU, see the\n[JAX quickstart](/tpu/docs/run-calculation-jax).\n| **Note:** If you aren't able to resolve your issue using this guide, see [Getting Support](/tpu/docs/getting-support) for further assistance.\n\nGeneral JAX issues\n------------------\n\nIf you run into issues while developing your training model or\ntraining with JAX, see the\n[JAX FAQ.](https://jax.readthedocs.io/en/latest/faq.html)\n\nFor more general programming\nerrors you might encounter when writing a training application with JAX, see\n[JAX Errors.](https://jax.readthedocs.io/en/latest/errors.html)\n\nProfiling JAX performance\n-------------------------\n\nYou can understand how your TPU resources are being utilized using the\ntools described in\n[Profiling JAX performance.](https://jax.readthedocs.io/en/latest/profiling.html)\n\nTroubleshooting memory issues\n-----------------------------\n\n| **Note:** TPU memory ([high bandwidth memory](/tpu/docs/system-architecture-tpu-vm#versions)) is shared across TensorCores on each TPU chip. This enables efficient coordination between the on-chip TensorCores.\n\nYou can monitor how the memory is used with the [JAX Device Memory Profiler](https://jax.readthedocs.io/en/latest/device_memory_profiling.html),\nbut you cannot directly manage how it is used.\n\nThe Device Memory Profiler can be used to:\n\n- Figure out [which arrays and executables are in TPU memory](https://jax.readthedocs.io/en/latest/device_memory_profiling.html#understanding-how-a-jax-program-is-using-gpu-or-tpu-memory) at a given time, or\n- [Track down memory leaks.](https://jax.readthedocs.io/en/latest/device_memory_profiling.html#debugging-memory-leaks)\n\nYou cannot specify how TPU memory is allocated for specific operations.\nFor more information on JAX-specific TPU performance issues, see\n[Performance Notes for using TPUs with JAX](/tpu/docs/run-calculation-jax#performance_notes).\n\nTroubleshooting TPU issues\n--------------------------\n\n### How can I verify that the TPU is running?\n\n**Details**\n\nEverything will be run on the TPU as long as JAX doesn't print\n\"No GPU/TPU found, falling back to CPU.\"\n\nYou can verify the TPU is active by either looking at `jax.devices()`, where\nyou should see several TPU devices displayed, or verify\nprogrammatically with: `assert jax.devices()[0].platform == 'tpu'`.\n\n### RuntimeError: Unable to initialize backend 'tpu': UNAVAILABLE: No TPU Platform available.\n\n**Details**\n\nThis runtime error message and/or finding the following in `/tmp/tpu_logs/tpu_driver.WARNING` on the TPU VM:\n`W1118 17:40:20.985243 23901 tpu_version_flag.cc:57] No hardware is found. Using default TPU version:xxxxxx`\ncan indicate that you are running the wrong TPU VM version.\n\nVerify that you are running the [current JAX runtime version](/tpu/docs/supported-tpu-versions#jax) and retry."]]
