runtime error

Cloning "https://github.com/google-research/big_vision" into "/tmp/big_vision_repo" Cloning into '/tmp/big_vision_repo'... Traceback (most recent call last): File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 879, in backends backend = _init_backend(platform) File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 970, in _init_backend backend = registration.factory() File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 668, in factory return xla_client.make_c_api_client(plugin_name, updated_options, None) File "/usr/local/lib/python3.10/site-packages/jaxlib/xla_client.py", line 200, in make_c_api_client return _xla.get_c_api_client(plugin_name, options, distributed_client) jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: No visible GPU devices. During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 15, in <module> import models File "/home/user/app/models.py", line 11, in <module> import paligemma_bv File "/home/user/app/paligemma_bv.py", line 25, in <module> mesh = jax.sharding.Mesh(jax.devices(), 'data') File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1082, in devices return get_backend(backend).devices() File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1016, in get_backend return _get_backend_uncached(platform) File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 995, in _get_backend_uncached bs = backends() File "/usr/local/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 895, in backends raise RuntimeError(err_msg) RuntimeError: Unable to initialize backend 'cuda': FAILED_PRECONDITION: No visible GPU devices. (you may need to uninstall the failing plugin package, or set JAX_PLATFORMS=cpu to skip this backend.)

Container logs: