Open imgarylai opened 6 years ago
https://github.com/deepmind/sonnet might be an example for building pip package to support both GPU and CPU version of tf simultaneously.
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.
You need mxnet gpu version.
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?