TensorFlow on MX150

One can run TensorFlow on NVidia GeForce MX150 graphics card using the following setup:

• CUDA version 8.0
• CuDNN version 6.0 for CUDA 8.0

Timing comparison for matrix multiplication using CPU (i7-8550) (shown in orange) and GPU (MX150) (shown in blue) for increasing matrix sizes.

 
{'/gpu:0': [0.23459196090698242, 0.5798056125640869, 0.08466935157775879, 0.0998234748840332], 
'/cpu:0': [0.7988660335540771, 4.3423473834991455, 14.469173431396484, 42.78021287918091]}

The code for doing the matrix multiplication is given here.