4 errors detected in the compilation of “render_utils_kernel.cu“

最新推荐文章于 2025-01-16 14:38:20 发布

原创最新推荐文章于 2025-01-16 14:38:20 发布 · 563 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#开发语言 #python #pytorch #三维重建 #3d

文章讲述了在Windows10平台上使用Python3.10.13和Torch1.12.1时，遇到在Cuda11.7环境中编译render_utils_kernel.cu文件时关于pow函数调用的错误，通过将pow函数从host到device的调用改为类型转换后解决了编译问题。

部署运行你感兴趣的模型镜像

部署环境：

平台：Windows 10

Python：3.10.13

Torch：1.12.1

Cuda：11.7

通过nvcc编译render_utils_kernel.cu的时候会返回以下错误信息

render_utils_kernel.cu(441): error: calling a __host__ function("std::conditional< ::std::is_same_v|| ::std::is_same_v, long double, ::std::conditional< ::std::is_same_v&& ::std::is_same_v, float, double> ::type> ::type ::pow<double, float, (int)0> (T1, T2)") from a __global__ function("raw2alpha_cuda_kernel<double> ") is not allowed

render_utils_kernel.cu(441): error: identifier "pow<double, float, (int)0> " is undefined in device code

render_utils_kernel.cu(515): error: calling a __host__ function("std::conditional< ::std::is_same_v|| ::std::is_same_v, long double, ::std::conditional< ::std::is_same_v&& ::std::is_same_v, float, double> ::type> ::type ::pow<double, float, (int)0> (T1, T2)") from a __global__ function("raw2alpha_backward_cuda_kernel<double> ") is not allowed

render_utils_kernel.cu(515): error: identifier "pow<double, float, (int)0> " is undefined in device code

4 errors detected in the compilation of "render_utils_kernel.cu".

根据错误提示，修改render_utils_kernel.cu的441行和515行

# 原始代码
alpha[i_pt] = 1 - pow((1 + e), (-interval));


# 修改为
alpha[i_pt] = 1 - pow((double)(1 + e), (double)(-interval));

# 原始代码
grad[i_pt] = min(exp_d[i_pt], 1e10) * pow((1+exp_d[i_pt]), (-interval-1)) * interval * grad_back[i_pt];

# 修改为

grad[i_pt] = min(exp_d[i_pt], 1e10) * pow((double)(1+exp_d[i_pt]), (double)(-interval-1)) * interval * grad_back[i_pt];

修改完这两行代码后，编译就可以顺利通过了

您可能感兴趣的与本文相关的镜像