使用CUDA/C++进行PyTorch扩展

• 英文文档

• 通过export CUDA_HOME指定路径，如果机器上有多个版本的CUDA

一般而言，和CUDA相关的报错可以通过设置CUDA版本(即CUDA路径)来实现。即export CUDA_HOME=

nvcc_args 编译时的参数设置

以可变形卷积为例进行说明
参考代码：?? https://github.com/haofeixu/aanet/tree/master/nets/deform_conv