AssertionError: Torch not compiled with CUDA enabled 未启用CUDA支持的PyTorch环境 中使用GPU加速
1.错误原因分析根本问题:当前安装的PyTorch是CPU版本,无法调用GPU硬件加速。当运行以下代码时会报错:model=YOLO("yolov8n.pt").to("cuda")#或.cuda()2.解决方案步骤步骤1:验证CUDA可用性在Python中运行以下命令检查环境:importtorchprint(torch.__version__)#查看PyTorch版本print(torch.c