Skip to content
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 support #69

Open
filchristou opened this issue Feb 15, 2024 · 2 comments
Open

GPU support #69

filchristou opened this issue Feb 15, 2024 · 2 comments

Comments

@filchristou
Copy link

filchristou commented Feb 15, 2024

Can a maintainer mention what is the state with the GPU support?
The README says that the gpu-support branch should be used, but that one was updated 5 years ago last time.
Given all the changes that happened in the meantime I guess it might be easier to make a new branch and support GPU from scratch. Also why a different branch? couldn't it be an option? It would be great if someone could provide some insight.

@zsunberg
Copy link
Member

Hi @filchristou . I think that, given the changes, the gpu-support branch is probably not viable any more. It would be a great contribution for someone to add it as an option!

@zengmao
Copy link

zengmao commented Sep 28, 2024

For the neural network part, is Flux's GPU support inherited by this package? Is it just the rest of the package that doesn't run on GPUs?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants