One can run TensorFlow on NVidia GeForce MX150 graphics card using the following setup:
The code for doing the matrix multiplication is given here.
- CUDA version 8.0
- CuDNN version 6.0 for CUDA 8.0
{'/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.