-
Notifications
You must be signed in to change notification settings - Fork 57
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.
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
GPU version is needed! #210
Comments
Hi! This code base uses T5X for training and inference on accelerators. Based on the note here https://github.com/google-research/t5x#installation T5X should work on GPUs. I'm not sure if it will automatically detect and use GPUs or if there is some flag that needs to be set. Hope that helps! |
I have tried the code on GPU and it's working. However, the cuda driver/toolkit and package setup takes some time. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi,
Thanks for reading my issue. I've learned from your amazing paper named The Power of Scale for Parameter-Efficient Prompt Tuning. I'm wondering whether I could train my prompt-based model with GPU on my PC, or must via Google Cloud TPU. In other words, could you please provide me with another version of codes which is only relied on GPU?
Looking forward to your reply!
Thanks
Chen
The text was updated successfully, but these errors were encountered: