search

ArtLabss / tennis-tracking

Open-source Monocular Python HawkEye for Tennis
https://www.artlabs.tech
The Unlicense
419 stars 100 forks source link

Use of GPU on Colab #30

Open fabiox23 opened 1 year ago

fabiox23 commented 1 year ago

Hi,

I have tried running the script on Colab following the instructions on the README file and installing the missing dependencies.

It does work but it doesn't seem to use the GPU despite the option being set up.

How can I ensure the GPU is used? Is it something to do with tensorflow? Any advice?

Kind regards.

shukkkur commented 1 year ago

@fabiox23 Hi!

Have you changed the runtime type to GPU? image

fabiox23 commented 1 year ago

Hi,

yes, that's the first thing I did. After running the script, I received a popup message saying the program wasn't using the GPU and asking whether I wanted to switch back to standard runtime. It took about 6 minutes to process a 3 second video so I think it wasn't using the GPU.

Does it need to import a different version of the tensorflow library and, if so, how do you specify that?

Thanks again.