环境:
ubuntu20.04
python3.8
torch1.12
cuda11.3
错误,编译sh build_and_install.sh,找不到cuda.h头文件
将所有项目文件下载好后,编译出现错误,找不到cuda.h头文件
解决方案:按照此文中的方法改代码。我是直接将他要改的文件都下载了下来,具体在他的博文中有介绍。
再进行编译,错误:#include <THC/THC.h>没有这个头文件
原因是THC.h在pytorch 1.12版本中被移除了,需要将头文件改一下。
在出现错误的代码中 注释掉i#include <THE/THC.h>
换成
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>
再编译就成功了