KOMURAVALLI NEERAJ

@waheed0332 But the recommended CUDA and Cudnn version is 10.0 and 7.4.x respectively with tensorflow 2.0.0. Also, what is the size of the data you are dealing with? What is...

@waheed0332 Thanks a lot, will give it a try and report back.

@waheed0332 So, I did try what you suggested and below are my findings: 1. So I first set up CUDA 10.1 and Cudnn 7.6 and then I built tensorflow(V2.3.1) from...

Here is the list of all packages with their respective versions Package Version ---------------------- --------- absl-py 0.10.0 astor 0.8.1 astunparse 1.6.3 boto3 1.16.3 botocore 1.19.3 cachetools 4.1.1 certifi 2020.6.20 chardet...

My Tensorflow-gpu was in fact 2.3.1 and as that did not work I reverted it back to 2.0.0. And my dataset has a images of from all ranges of size...

Will add test cases In `test_image_type.py` and push all the modifications you recommended in the next push.