apex安装报错:Cuda extensions are being compiled with a version of Cuda that does not match the verson

apex是nvdia的半精度运算库,能够增加运算速度,减小现存占用。
apex的仅支持带有tensor core的GPU,即RTX显卡。

apex的安装方式如下:

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

但是我在安装过程中:报错了

RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 9.0.176.

看这个意思是说cuda跟pytorch的版本不对应。后来我看到apex直接不带–global-option安装好像也能用。
于是修改最后一句

pip install -v --no-cache-dir ./

安装成功

  • 31
    点赞
  • 37
    收藏
    觉得还不错? 一键收藏
  • 16
    评论
runtimeerror: cuda extensions are being compiled with a version of cuda that is different from the version installed. 该错误是由于使用的CUDA编译器版本与已安装CUDA版本不一致导致的。CUDA是一种用于并行计算的平台和编程模型,用于在GPU上执行高性能计算任务。在进行CUDA扩展编译时,必须使用与已安装CUDA版本匹配的编译器。 要解决这个问题,首先需要确定已安装CUDA的版本。可以使用命令行工具`nvcc --version`来获取CUDA版本信息。确保安装CUDA版本与要编译的CUDA扩展要求的版本一致。 如果已安装CUDA版本与要编译的扩展不一致,需要卸载当前版本的CUDA,然后安装与扩展要求的版本相匹配的CUDA。可以从NVIDIA官方网站下载正确版本的CUDA安装包,并按照安装指南进行安装。 另外,还需要确保将确定要使用的CUDA版本设置为系统环境变量中的默认版本。这可以通过修改系统的`PATH`环境变量来实现。在环境变量中将CUDA安装路径添加到`PATH`中,以便操作系统可以找到正确的CUDA版本。 在确保CUDA版本匹配的情况下,重新编译CUDA扩展。根据具体的编译方式,需要使用适当的命令和选项来编译和链接CUDA扩展。在编译过程中,确保使用正确版本的CUDA编译器和库。 总结来说,修复这个错误的关键是确保CUDA扩展的编译器版本与已安装CUDA版本相匹配,并设置正确的环境变量来指定要使用的CUDA版本。通过这些步骤,可以成功解决这个问题并正确编译CUDA扩展。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值