How can I run ImageAI on my GPU and not my CPU

Question

So I am quite new to this I was trying to find answers on google but it is kind of not working. So I am trying to run this library ImageAI library

I am able to run it normally on the CPU at least I think it runs on CPU by just calling python test.py. Am I correct there

But since the model prediction takes a long time I would like to run it on my GPU. What I was trying to do is create a conda environment and activate it but after I do I get this error.

ModuleNotFoundError: No module named 'imageai.Classification'

Although I have imageai installed in my environment

pip freeze | findstr imageai
imageai==2.1.5

As you can see by executing this command. What am I doing wrong here?

Answer 1

I found the solution it doesnt require the conda environment. ImageAI automatically runs on GPU if available. All you need to do is to ensure you have the GPU version of Tensorflow installed.