New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Unable to initialize backend 'tpu': UNIMPLEMENTED: LoadPjrtPlugin is not implemented on windows yet. (set JAX_PLATFORMS='' to automatically choose an available backend) #40

Open

TeaMaster1337 opened this issue Mar 18, 2024 · 1 comment

TeaMaster1337 commented Mar 18, 2024

How to fix that?

ngovanuc commented Apr 16, 2024

import os
os.environ['JAX_PLATFORMS'] = ''

add the above code.
If you're using Jupyter notebook try to run all cells in one time.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment