Запуск программы на пайтон через GPU

Рейтинг: 0Ответов: 0Опубликовано: 27.06.2023

Я запускаю программу, работающую на основе библиотеки 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.

Ответы

Ответов пока нет.