Tensorflow with gpu while submitting

Hi,
would someone knows how to get tensorflow running on gpu while submitting.
gpu is true in the json file. tensorflow is running as I m getting metrics in a few submissions but I am getting :

W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory;