Запуск программы на пайтон через GPU
Я запускаю программу, работающую на основе библиотеки easyocr, при каждом запуске программа выдает в консоль информацию:
Neither CUDA nor MPS are available - defaulting to CPU.
Примечание: Этот модуль работает намного быстрее с графическим процессором.
Теперь я пытаюсь запустить программу через gpu, я установил cuda v.11.8, но при запуске программы print(f"CUDA version: {torch.version.cuda}")
:
if torch.cuda.is_available():
device = torch.device('cuda')
print('Using GPU.')
else:
device = torch.device('cpu')
print('Using CPU.')
Выдает:
CUDA version: None Using CPU.
Подскажите, что нужно исправить, для того, чтобы python видел установленную версию cuda, а также запускал программу через gpu.
Источник: Stack Overflow на русском