as written by prateek bhaiya. model=(weight=“imagenet”)
i am getting the error in Cudagetdevice().
I am facing the error in creating the model in transfer extract
This error occurs due to wrong installation of gpu drivers, make sure you have installed tensorflow cpu and not tensorflow gpu. If you have installed tensorflow gpu than uninstall it using pip and again install tensorflow cpu.
If error still persists, share the screenshot of the error.