yes! if I try running this on the CPU, the whole system freezes to the point where I have to manually restart the computer. Also if I try running the code with lower image resolution, lower batch sizes etc, each epoch takes around 12 hours to complete on a CPU which is definitely impractical.
Now…about using other GPU environments, I used the google colab to train the model, but it shows an error after a few hours of training saying ‘page not responsive’. I thought about colab pro (which is a paid version of colab) but it is available only in the United States…and I am from a different country.  I am also thinking about aws EC2 p3 instances for renting a cloud GPU…but again, I don’t really have sufficient documentation about it’s usage…or how it is going to help me to create a deep learning website.
I do have a question though. Suppose I somehow successfully train the model on colab and save the model as the bin file (just like how I did in the code above), is it possible to download that trained model file to my local directory for prediction without the gpu? My question maybe complicated…apologies. I am just trying different ways.