How to train with GPU

[JJY0710]

[sjtu-cz]

I have the same question

[dvivarelli]

If you have CUDA 11.2 like me, you should upgrade dependencies Keras==2.4.3 tensorflow==2.5.0 (tested with python 3.7 and it works)

update dipendencies (tested on Colab):

astor==0.8.0
cycler==0.10.0
h5py~=3.1.0
Keras==2.4.3
kiwisolver==1.1.0
Markdown==3.1.1
matplotlib==3.1.1
opencv-python==4.1.1.26
protobuf==3.20
pyparsing==2.4.2
python-dateutil==2.8.0
PyYAML==5.1.2
scipy==1.3.1
tensorboard==2.5.0
tensorflow==2.5.0
termcolor==1.1.0
Werkzeug==0.16.0
wget==3.2

I think, it should work also with CUDA 10 (without upgrade dipendencies)