Using multi-GPU framework you can run any normal tensorflow code on a multiple specified number of GPUs. The framework creates same copy of the tensorflow graph for each gpu and run parallely. The framework does not require any major change in the design of the tf graph. You can write your tensorflow code any way you want and use framework method call to enable it on multiple GPUs.
This project implements LSTM and CNN using the Multi-GPU framework on MNIST data.
To run this project you will need following packages:
- TensforFlow
- NumPy
- Manish Vidyasagar
This project is licensed under the MIT License - see the LICENSE.md file for details