Итак, я относительно новичок во всем этом, и мне нужно запрограммировать анализ походки для нескольких людей с помощью Yolov7. Предполагается, что программа использует графический процессор компьютера, но может обнаружить только процессор. Моя версия Cuda — 12.2, а модель графического процессора — NVIDIA Geforce RTX 2080 Ti. Могу ли я узнать, в чем возможная проблема? Будь то коды или ПК.
"" Python
# Import necessary libraries
import torch
# Function to load YOLOv7 model
def load_yolov7_model(model_path='yolov7-w6-pose.pt'):
"""
Function to load the YOLOv7 model.
Args:
model_path (str): Path to the model file (default: 'yolov7-w6-pose.pt').
Returns:
yolo_model: Loaded YOLOv7 model.
device: The device where the model is loaded (GPU or CPU).
"""
try:
# Check if GPU is available and set the device
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(f"Loading model on device: {device}")
# Load the YOLOv7 model
yolo_model = torch.load(model_path, map_location=device)['model']
yolo_model.float().eval() # Set to evaluation mode
# Move model to GPU if available
if torch.cuda.is_available():
yolo_model.half().to(device)
print("Model loaded successfully.")
return yolo_model, device
except Exception as e:
print(f"Error loading model: {e}")
return None, None
# Example usage
if __name__ == "__main__":
model_path = 'yolov7-w6-pose.pt' # Ensure the path to your model file is correct
model, device = load_yolov7_model(model_path)
if model is None:
print("Failed to load model.")
else:
print("Model ready for inference.")
Подробнее здесь: https://stackoverflow.com/questions/789 ... l-computer
