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

Two distributions need. CPU and GPU version #16

Open
imgarylai opened this issue Oct 5, 2017 · 3 comments
Open

Two distributions need. CPU and GPU version #16

imgarylai opened this issue Oct 5, 2017 · 3 comments
Assignees
Labels

Comments

@imgarylai
Copy link
Member

imgarylai commented Oct 5, 2017

The problem is tensorflow has cpu and gpu distributions. If we want to support both solution, we also have to create two distributions, to resolve the dependency. Do we want to support both solutions?

@imgarylai
Copy link
Member Author

https://github.com/deepmind/sonnet might be an example for building pip package to support both GPU and CPU version of tf simultaneously.

@Jeevi10
Copy link

Jeevi10 commented Mar 9, 2020

does current version support gpu?
I have been trying to decode a R8 news dataset and it takes forever.
I have Nvidia 1080 Ti GPU and 64 gb ram.

@hankcs
Copy link
Contributor

hankcs commented Mar 10, 2020

You need mxnet gpu version.

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

No branches or pull requests

4 participants