@akash18014 @picekl From what I remember, we have to explicitly set only cuda version 10.0 (Can I have an example of a code which is working to make a submission on gitlab?) . Maybe downgrade cuda toolkit 10.1 to 10.0 and make sure corresponding tensorflow works with 10.0.
This is the only link which helped me downgrade (https://dmitry.ai/t/topic/33)