Jamshaid Sohail
i am also the same issue on yolo v5. 
Hi. I am facing the same issue on my local PC with GeForce RTX 3060 (6GB) but when I run the code on Goolge Colab, everything seems to work very...
Hey I resolved the issue by upgrading the tensorflow version to 2.5. pip install tensorflow-gpu==2.5 The above command did the job for me. Good luck.