Two distributions need. CPU and GPU version

Open imgarylai opened this issue 8 years ago • 3 comments

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?

Oct 05 '17 15:10 imgarylai

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

Oct 05 '17 15:10 imgarylai

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.

Mar 09 '20 23:03 Jeevi10

You need mxnet gpu version.

Mar 10 '20 00:03 hankcs