The Graphical processing unit (GPU) is hardware used for rendering images using a lot of cores. Pascal is the latest GPU micro architecture released by NVIDIA. The presence of hundreds of cores in GPU helps enhance the computation. This section provides the recipe for running a deep learning model using GPU.
Training a deep learning model on a GPU
Getting ready
This section provides dependencies required to run GPU and CPU:
- The experiment performed in this recipe uses GPU hardware such as GTX 1070.
- Install mxnet for GPU. To install mxnet for GPU for a specified machine, follow the installation instruction from mxnet.io. Select the requirement as shown in the screenshot, and follow the instructions:
Steps to get installation...