Я использую rembg для удаления фона с изображений. Сейчас одна активация функции удаления занимает около 0,13 с. Но мне нужно, чтобы это было быстрее. Поэтому я решил, что использование CUDA gpu для запуска процесса удаления ускорит процесс, но я не понимаю, как заставить сессию использовать CUDA?
P.s. Если я распечатаю session.inner_session.get_providers(), это покажет, что доступны как поставщик выполнения CUDA, так и поставщики выполнения CPU.
Подробнее здесь: https://stackoverflow.com/questions/782 ... d-in-rembg
