Pytorch extension-cuda

error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

方法：

将.cu文件中的 #include <torch/extension.h> 改为 #include <torch/types.h>