Clarification: Does TinyGrad use CUDA? #841
-
Just a clarification as I was unsure, does TinyGrad support CUDA for GPU acceleration? |
Beta Was this translation helpful? Give feedback.
Answered by
jla524
Jun 3, 2023
Replies: 1 comment 2 replies
-
It does. You can install dependencies for CUDA by running |
Beta Was this translation helpful? Give feedback.
2 replies
Answer selected by
sam-the-programmer
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It does. You can install dependencies for CUDA by running
pip install -e ".[cuda]"
, and your default device will be set to CUDA.