Replies: 1 comment
-
PS: libtpu has been removed from python for being empty and there is no code in the github repository. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I've been battling with Debian and have managed to clear most of the issues... or at least got to where grok starts running for long enough to get into a debugger.
Could some kind soul tell me how to fix these?
`INFO:jax._src.xla_bridge:Unable to initialize backend 'cuda':
INFO:jax._src.xla_bridge:Unable to initialize backend 'rocm': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig'
INFO:jax._src.xla_bridge:Unable to initialize backend 'tpu': INTERNAL: Failed to open libtpu.so: libtpu.so: cannot open shared object file: No such file or directory
INFO:rank:Initializing mesh for self.local_mesh_config=(1, 8) self.between_hosts_config=(1, 1)...
INFO:rank:Detected 1 devices in mesh
ValueError: Number of devices 1 must equal the product of mesh_shape (1, 8)`
Beta Was this translation helpful? Give feedback.
All reactions