Блог

Когда я загружаю модель с помощью YOLOv5 и указываю 'cpu' , что yolo все еще используется 'gpu' . Почему?

Код:

 model = yolov5.load(model, device='cuda' if torch.cuda.is_available() else 'cpu')  

 print('cuda' if torch.cuda.is_available() else 'cpu')  

С принтами cpu

Он всегда пишет следующее сообщение в консоль, что довольно раздражает:

 UserWarning: User provided device_type of 'cuda', but CUDA is not available. Disabling  warnings.warn('User provided device_type of 'cuda', but CUDA is not available. Disabling')