Tensorflow-GPU in RAPIDS for Neural Networks to do Computer Vision

Accelerated Computing GPU-Accelerated Libraries
1

I figured that cuML does not have functions for Neural Networks, so I am wondering if it is possible to use tensorflow-gpu in RAPIDS. I installed tf-gpu in RAPIDS container but tf does not see the GPUs I checked it with

tf.test.is_gpu_available()

function and it returns FALSE. I am not sure what I am doing wrong.

2

Found out that we need to be using -dev version of the container.