Use of GPU on Colab

[fabiox23]

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]

@fabiox23 Hi!

Have you changed the runtime type to GPU?

[fabiox23]

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.