DeepImageSearch icon indicating copy to clipboard operation
DeepImageSearch copied to clipboard

GPU not used

Open igorpanjan-redbumerang opened this issue 1 year ago • 7 comments

I've uninstalled faiss-cpu and installed faiss-gpu as you mentioned in the docs. But when I run st.run_index() GPU is not utilized.

GPU utilization is very important to me, because I have to train cca. 500 GB of images. My GPU is Nvidia GTX 1080 Ti.

Nvidia drivers are installed correctly - Nvidia-smi returns list of GPUs. I've also run torch.cuda.is_available(), which returns True.

I would appreciate any pointers.

I also have a feature request - match batch of similar images.

Library is great BTW!

igorpanjan-redbumerang avatar Apr 28 '23 08:04 igorpanjan-redbumerang