weixin_57731382的博客

原创 50系显卡导出onnx模型异常解决办法

摘要：在更换50系显卡并重新安装PyTorch和CUDA后，模型训练正常但导出ONNX时出现0xC0000005错误。经排查发现该问题与GPU无关，而是由于使用了PyTorch 2.10.0开发版导致的底层C++错误。解决方案是创建新的稳定环境（Python 3.10 + PyTorch 2.1.2 + CUDA 12.1），在该环境下成功导出了ONNX模型。这表明开发版PyTorch在ONNX导出阶段存在稳定性问题，建议将训练和导出环境分离。

原创 50系显卡安装pytorch避坑笔记

50系显卡安装PyTorch解决方案：针对RTX5090/5080等采用AdaLovelaceNext架构的新显卡，需全新安装PyTorch环境。具体步骤包括：1）彻底删除旧环境；2）创建Python3.10纯净环境；3）安装PyTorchNightly版本(cu130)；4）验证CUDA可用性；5）安装YOLOv8基础库。该方法成功解决了新显卡架构兼容性问题，验证结果显示RTX5070显卡可正常训练模型。注意必须使用最新PyTorch版本才能支持9.0及以上的计算能力。

原创 科技与美学的完美结合：硬件连接艺术