ARM64平台编译libtorch

编译过程：（参考链接）

执行python ../tools/build_libtorch.py这一步时报错：

Traceback (most recent call last):
  File "../tools/build_libtorch.py", line 14, in <module>
    from tools.build_pytorch_libs import build_caffe2
  File "/home/Thirdparty/pytorch/tools/build_pytorch_libs.py", line 10, in <module>
    from .setup_helpers.cudnn import CUDNN_INCLUDE_DIR, CUDNN_LIBRARY, USE_CUDNN
  File "/home/Thirdparty/pytorch/tools/setup_helpers/cudnn.py", line 52, in <module>
    raise AssertionError("Could not find #define CUDNN_MAJOR in " + include_file_path)
AssertionError: Could not find #define CUDNN_MAJOR in /usr/include/cudnn.h

问题解决：

打开/pytorch/tools/setup_helpers/cudnn.py这个文件

修改第42行

'cudnn.h'改成'cudnn_version.h'

问题解决

原理：cudnn 8的版本，将有版本号的头文件单独写了一个文件cudnn_version.h，而不再是之前的cudnn.h（参考链接）