问题:
在Torch1.10环境下,build含CUDA算子编译的工程时出错:
build命令:
python setup.py build develop
错误:
ATen/ATen.h: No such file or directory
分析:
出错的主要原因是用于编译算子的torch版本不合适,升级Torch1.11后正常编译。
在Torch1.10环境下,build含CUDA算子编译的工程时出错:
build命令:
python setup.py build develop
错误:
ATen/ATen.h: No such file or directory
出错的主要原因是用于编译算子的torch版本不合适,升级Torch1.11后正常编译。