【CUDA】算子编译错误ATen/ATen.h: No such file or directory

问题:

在Torch1.10环境下,build含CUDA算子编译的工程时出错:

build命令:

python setup.py build develop

错误:

ATen/ATen.h: No such file or directory

分析:

出错的主要原因是用于编译算子的torch版本不合适,升级Torch1.11后正常编译。

你可能感兴趣的:(AI/ML/DL,人工智能,深度学习)