Python Deep Learning Cookbook

How to do it...

1. To install MXNet on Ubuntu with GPU support, we can use the following command in the terminal:

pip install mxnet-cu80==0.11.0

For other platforms and non-GPU support, have a look at https://mxnet.incubator.apache.org/get_started/install.html.

2. Next, we are ready to import mxnet in our Python environment:

import mxnet as mx

3. We create some simple dummy data that we assign to the GPU and CPU:

import numpy as npx_input = mx.nd.empty((1, 5), mx.gpu())x_input[:] = np.array([[1,2,3,4,5]], np.float32)y_input = mx.nd.empty((1, 5), mx.cpu())y_input[:] = np.array([[10, 15, 20, 22.5, 25]], np.float32)

4. We can easily copy and adjust the data. Where possible MXNet will automatically execute operations in parallel:

x_inputw_input = ...