Apex安装填坑

第一个坑:cuda版本与pytorch构建版本不一致

因为服务器有很多人在使用,所以会出现cudnn构建版本与pytorch本身版本不一致的现象,使用conda环境安装的缺少编译apex所需的组件,这一点会造成apex安装不成功。这一点可以在报错信息里找到下面类似的提示:

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 xxxx.

解决方法

这种方法需要去cuda的官网下载与pytorch相对应的版本,安装成功后修改CUDA_HOME环境继续编译程序即可解决问题。

第二个坑:GCC版本构建问题

构建时候会报错提示GCC版本不支持,这个也简单,在官网查一下当前cuda支持的gcc版本,重新去装一个就成,可以解决问题,就是编译gcc的时间巨长,需要耐心

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值