No GPU Using
DH-HAN opened this issue · comments
I1025 08:07:07.369448 139768635004736 xla_bridge.py:357] Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker:
I1025 08:07:07.369699 139768635004736 xla_bridge.py:357] Unable to initialize backend 'cuda': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig'
I1025 08:07:07.369804 139768635004736 xla_bridge.py:357] Unable to initialize backend 'rocm': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig'
I1025 08:07:07.370839 139768635004736 xla_bridge.py:357] Unable to initialize backend 'tpu': INVALID_ARGUMENT: TpuPlatform is not available.
I1025 08:07:07.370978 139768635004736 xla_bridge.py:357] Unable to initialize backend 'plugin': xla_extension has no attributes named get_plugin_device_client. Compile TensorFlow with //tensorflow/compiler/xla/python:enable_plugin_device set to true (defaults to false) to enable this.
W1025 08:07:07.371087 139768635004736 xla_bridge.py:364] No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
I set up my environment following your guide, inclue jax, jax lib but
This kind of error message come out, and i automatically inference image by using only gpu
are there someone who have same issue?
I solve this issue by installing Right version of JAX thank you for your nice sharing!