Spaces:
Runtime error
error
Exit code: 1. Reason: ges/jax/_src/util.py", line 256, in cached
return f(*args, **kwargs)
File "/usr/local/lib/python3.10/site-packages/jax/_src/dispatch.py", line 222, in xla_primitive_callable
compiled = _xla_callable_uncached(
File "/usr/local/lib/python3.10/site-packages/jax/_src/dispatch.py", line 249, in _xla_callable_uncached
computation = sharded_lowering(
File "/usr/local/lib/python3.10/site-packages/jax/_src/dispatch.py", line 241, in sharded_lowering
return pxla.lower_sharding_computation(
File "/usr/local/lib/python3.10/site-packages/jax/_src/profiler.py", line 314, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/site-packages/jax/_src/interpreters/pxla.py", line 1914, in lower_sharding_computation
backend, device_assignment = _get_and_check_device_assignment(
File "/usr/local/lib/python3.10/site-packages/jax/_src/interpreters/pxla.py", line 1680, in _get_and_check_device_assignment
final_device_assignment = (_get_default_device(),)
File "/usr/local/lib/python3.10/site-packages/jax/_src/interpreters/pxla.py", line 1645, in _get_default_device
return config.jax_default_device or xb.local_devices()[0]
File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 808, in local_devices
process_index = get_backend(backend).process_index()
File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 710, in get_backend
return _get_backend_uncached(platform)
File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 691, in _get_backend_uncached
bs = backends()
File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 628, in backends
raise RuntimeError(err_msg)
RuntimeError: Unable to initialize backend 'tpu': INTERNAL: Failed to open /usr/local/lib/python3.10/site-packages/libtpu/libtpu.so: /usr/local/lib/python3.10/site-packages/libtpu/libtpu.so: cannot open shared object file: No such file or directory (set JAX_PLATFORMS='' to automatically choose an available backend)