DeepImageSearch
DeepImageSearch copied to clipboard
GPU not used
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!