Проблемы с использованием графического процессора школьного компьютера ⇐ Python

[Anonymous]

Итак, я относительно новичок во всем этом, и мне нужно запрограммировать анализ походки для нескольких людей с помощью Yolov7. Программа работает нормально, однако программа не использует графический процессор компьютера. Это код, который я сейчас использую, чтобы использовать графический процессор. Я использую виртуальную среду (Anaconda3), чтобы иметь все библиотеки и уже установить факел и факельное зрение. Моя версия Cuda — 12.2, а модель графического процессора — NVIDIA Geforce RTX 2080 Ti. Может ли кто-нибудь помочь мне с моей проблемой.
Код для использования компьютерного графического процессора вместо процессора:
`` def load_yolov7_model(model_path='yolov7-w6-pose.pt') : устройство = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") yolo_model = torch.load(model_path, map_location=device)['model'] yolo_model.float().eval( ) if torch.cuda.is_available(): yolo_model.half().to(device) возвращает yolo_model, устройство

Код: Выделить всё

Когда код выполняется, программа будет использовать ЦП вместо графического процессора.
Я использовал этот код, чтобы дважды проверить, использует ли он ЦП:
``import torch if torch.cuda.is_available() == True: print("ГП используется") else: print("ЦП используется")

Код: Выделить всё

Чтобы убедиться, что он использует процессор
Я также провел стресс-тест графического процессора, чтобы проверить, правильно ли он работает, и так оно и было. Не слишком уверен, что еще я могу сделать.


Подробнее здесь: https://stackoverflow.com/questions/789 ... l-computer