remove fancy CUDA device selection, for plain 'cuda'

retrieval.py

@@ -9,12 +9,12 @@ from langchain.vectorstores import Pinecone
 
 # load custom code
 from clip_for_ppts import ClipImage
-from gpu_memory_utils import (get_device_with_most_free_memory,
-                              get_free_memory_dict,
-                              get_gpu_ids_with_sufficient_memory)
 
-# from PIL import Image
+# from gpu_memory_utils import (get_device_with_most_free_memory,
+#                               get_free_memory_dict,
+#                               get_gpu_ids_with_sufficient_memory)
 
+# from PIL import Image
 
 LECTURE_SLIDES_DIR = os.path.join(os.getcwd(), "lecture_slides")
 PINECONE_API_KEY = os.environ.get("PINECONE_API")
@@ -81,7 +81,7 @@ class Retrieval:
     self.clip_search_class = ClipImage(path_of_ppt_folders=LECTURE_SLIDES_DIR,
                                        path_to_save_image_features=os.getcwd(),
                                        mode='text',
-                                       device=f'cuda:{get_device_with_most_free_memory()}')
+                                       device='cuda')
 
   def reverse_img_search(self, img):
     imgs = self.clip_search_class.image_to_images_search(img)