Is there a way to use a GPU only when training an Index  in the process of add_faisis_index?
{'login': 'shamanez', 'id': 16892570, 'node_id': 'MDQ6VXNlcjE2ODkyNTcw', 'avatar_url': 'https://avatars.githubusercontent.com/u/16892570?v=4', 'gravatar_id': '', 'url': 'https://api.github.com/users/shamanez', 'html_url': 'https://github.com/shamanez', 'followers_url': 'https://api.github.com/users/shamanez/followers', 'following_url': 'https://api.github.com/users/shamanez/following{/other_user}', 'gists_url': 'https://api.github.com/users/shamanez/gists{/gist_id}', 'starred_url': 'https://api.github.com/users/shamanez/starred{/owner}{/repo}', 'subscriptions_url': 'https://api.github.com/users/shamanez/subscriptions', 'organizations_url': 'https://api.github.com/users/shamanez/orgs', 'repos_url': 'https://api.github.com/users/shamanez/repos', 'events_url': 'https://api.github.com/users/shamanez/events{/privacy}', 'received_events_url': 'https://api.github.com/users/shamanez/received_events', 'type': 'User', 'site_admin': False}
[{'id': 1935892912, 'node_id': 'MDU6TGFiZWwxOTM1ODkyOTEy', 'url': 'https://api.github.com/repos/huggingface/datasets/labels/question', 'name': 'question', 'color': 'd876e3', 'default': True, 'description': 'Further information is requested'}]
[]
Motivation - Some FAISS indexes like IVF consist of the training step that clusters the dataset into a given number of indexes. It would be nice if we can use a GPU to do the training step and covert the index back to CPU as mention in [this faiss example](https://gist.github.com/mdouze/46d6bbbaabca0b9778fca37ed2bcccf6).
{'url': 'https://api.github.com/repos/huggingface/datasets/issues/2108/reactions', 'total_count': 0, '+1': 0, '-1': 0, 'laugh': 0, 'hooray': 0, 'confused': 0, 'heart': 0, 'rocket': 0, 'eyes': 0}