fatal error: ATen/cuda/DeviceUtils.cuh: No such file or directory

在编译apex的时候,遇到以下报错。
查询百度无果,查询谷歌一次得到结果。还是那句话,有问题找谷歌,别找百度。

具体原因:缺少CUDA头文件CUDAGraphsUtils.cuh。
而这个文件是从torch==1.8.0a后开始加入的,所以如果你的pytorch版本是1.7、1.6、1.5什么的那肯定没有。
解决方法就是回滚apex的版本。
apex不是软件,回滚需要通过git操作来进行:

git reset --hard 3fe10b5597ba14a748ebb271a6ab97c09c5701ac

在apex根文件夹内,运行以上指令,再执行编译即可。

附上apex下载并编译指令:

git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext
  • 10
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值