JupyterHub - GPU Notebooks

I have a K8s environment with GPU nodes, the drivers are installed on the node. I have built the Jupyterhub Notebook image which i am spawning. After i spawn i don’t see the GPU in the available devices. Do i also have to install any specific drivers in my Jupyter Notebook Image as well ? If not, how can i leverage GPU in my Jupyter Notebok container ?

Tensorflow versions being used:

  • tensorflow==1.14.0
  • tensorflow-gpu==1.14.0

CUDA Driver version 10.0