Pytorch extension-cuda

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

方法:

将.cu文件中的 #include 改为 #include

你可能感兴趣的:(Pytorch extension-cuda)