炼丹日记00-RTX4070 windows系统跑通MMPose示例代码(解决了ValueError: numpy.dtype size changed,...)

MMPose RTX 40系列GPU上的终极部署方案 (解决了ValueError: numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject问题)

背景提要

本方案旨在解决使用较新NVIDIA显卡(如RTX 4070, 计算能力sm_89)时,因旧版CUDA/PyTorch不兼容而导致的一系列连锁依赖问题。最终目标是在一个全新的Docker环境中,从零开始完整部署并成功运行MMPose的图像推理demo。

解决了配置MMPose过程中的什么问题?

  1. MMPose官方docker镜像cuda版本过低,无法支持RTX 40系列架构GPU的问题
  2. 示例代码缺少运行系统库问题
  3. numpy和xtcocotools的二进制冲突问题 (最关键的问题,也是解决起来最有技术含量的问题:ValueError: numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject)

部署步骤

第一步:环境准备 - 构建一个现代化的Docker容器

问题的根源是官方docker镜像的cuda本本过低无法驱动40系GPU。因此,第一步是拉取一个预装了高版本CUDA和PyTorch的Docker镜像。

  1. 拉取Docker镜像 (此镜像内置CUDA 11.8和PyTorch 2.3.0):

    docker pull pytorch/pytorch:2.3.0-cuda11.8-cudnn8-devel
    
  2. 配置并启动容器:
    修改你的 docker-compose.ymldocker run 命令,将

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值