Nvidia apex模型量化库安装

本文能解决的问题

安装途中编译报错,安装好了某个版本后,代码跑起来的途中因为apex报错(诸如:fused cuda的错误、其他各类module not found的错误、cannot import name ‘container_abcs’ from 'torch._six’等等)。

重点:找对版本号

在你排除了所有的“英伟达驱动未安装”,“驱动与cuda不匹配”等前置问题,并能顺利进入安装流程后,若出现文章开头提到的问题,那么可以到apex的源码处使用如下命令切换git分支

git checkout 2386a912164b0c5cfcd8be7a2b890fbac5607c82

这个是目前能找到的最新的能用且无bug的代码版本,然后通过官方提供的安装命令进行安装(不要用其他博客说的,类似python setup.py install --cpp_ext --cuda_ext 这种通过直接python运行setup代码的安装,因为仔细阅读了安装流程,英伟达明确说了这个安装方式被废弃了,实测结果就是会导致cuda cpp那两东西没装上!)
官方代码:

pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
  • 10
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值