error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
方法：
将.cu文件中的 #include <torch/extension.h> 改为 #include <torch/types.h>
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1
方法：
将.cu文件中的 #include <torch/extension.h> 改为 #include <torch/types.h>
本文标题：Pytorch extension-cuda
本文链接：https://www.haomeiwen.com/subject/gmofdctx.html
网友评论