cuda nvcc生成ptx文件遇到的错误error: invalid redeclaration of type name "size_t"

博客给出解决问题的方法,使用vs的x64本机命令提示符可避免某错误,且经过亲测有效。
部署运行你感兴趣的模型镜像

解决方法:用vs的x64本机命令提示符可避免该错误,亲测有效!

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

PyTorch 2.7

PyTorch 2.7

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

2025-09-18 14:17:00.603094: W tensorflow/compiler/mlir/tools/kernel_gen/tf_gpu_runtime_wrappers.cc:40] 'cuModuleLoadData(&module, data)' failed with 'CUDA_ERROR_INVALID_PTX' 2025-09-18 14:17:00.603148: W tensorflow/compiler/mlir/tools/kernel_gen/tf_gpu_runtime_wrappers.cc:40] 'cuModuleGetFunction(&function, module, kernel_name)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-18 14:17:00.603193: W tensorflow/core/framework/op_kernel.cc:1827] INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-18 14:17:00.603235: W tensorflow/core/framework/local_rendezvous.cc:404] Local rendezvous is aborting with status: INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [2025-09-18 14:17:00,603][pinnstf2.utils.utils][ERROR] - Traceback (most recent call last): File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/common/variables.py", line 153, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/core.py", line 69, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/core.py", line 139, in convert_to_tensor return tf.cast(x, dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/tensorflow/python/framework/ops.py", line 5983, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: [2025-09-18 14:17:00,609][pinnstf2.utils.utils][INFO] - Output dir: /home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/examples/aneurysm3D/outputs/14-16-46 Error executing job with overrides: [] Traceback (most recent call last): File "/home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/train.py", line 139, in main metric_dict, _ = pinnstf2.train( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/utils/utils.py", line 84, in wrap raise ex File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/common/variables.py", line 153, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/core.py", line 69, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/keras/src/backend/tensorflow/core.py", line 139, in convert_to_tensor return tf.cast(x, dtype) File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.10/site-packages/tensorflow/python/framework/ops.py", line 5983, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
09-19
2025-09-20 18:11:55.609791: W tensorflow/compiler/mlir/tools/kernel_gen/tf_gpu_runtime_wrappers.cc:40] 'cuModuleLoadData(&module, data)' failed with 'CUDA_ERROR_INVALID_PTX' 2025-09-20 18:11:55.609846: W tensorflow/compiler/mlir/tools/kernel_gen/tf_gpu_runtime_wrappers.cc:40] 'cuModuleGetFunction(&function, module, kernel_name)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-20 18:11:55.609887: W tensorflow/core/framework/op_kernel.cc:1842] INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-20 18:11:55.609927: I tensorflow/core/framework/local_rendezvous.cc:407] Local rendezvous is aborting with status: INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [2025-09-20 18:11:55,610][pinnstf2.utils.utils][ERROR] - Traceback (most recent call last): File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( ^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() ^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/common/variables.py", line 173, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 73, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 152, in convert_to_tensor return tf.cast(x, dtype) ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/framework/ops.py", line 6027, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: [2025-09-20 18:11:55,614][pinnstf2.utils.utils][INFO] - Output dir: /home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/outputs/18-11-42 Error executing job with overrides: [] Traceback (most recent call last): File "/home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/train.py", line 139, in main metric_dict, _ = pinnstf2.train( ^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 84, in wrap raise ex File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( ^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() ^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/common/variables.py", line 173, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 73, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 152, in convert_to_tensor return tf.cast(x, dtype) ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/framework/ops.py", line 6027, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
最新发布
09-21
cuModuleLoadData(&module, data)' failed with 'CUDA_ERROR_INVALID_PTX' 2025-09-19 00:59:41.652223: W tensorflow/compiler/mlir/tools/kernel_gen/tf_gpu_runtime_wrappers.cc:40] 'cuModuleGetFunction(&function, module, kernel_name)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-19 00:59:41.652257: W tensorflow/core/framework/op_kernel.cc:1842] INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' 2025-09-19 00:59:41.652285: I tensorflow/core/framework/local_rendezvous.cc:407] Local rendezvous is aborting with status: INTERNAL: 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [2025-09-19 00:59:41,652][pinnstf2.utils.utils][ERROR] - Traceback (most recent call last): File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( ^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() ^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/common/variables.py", line 173, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 73, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 152, in convert_to_tensor return tf.cast(x, dtype) ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/framework/ops.py", line 6027, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: [2025-09-19 00:59:41,657][pinnstf2.utils.utils][INFO] - Output dir: /home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/outputs/00-59-28 Error executing job with overrides: [] Traceback (most recent call last): File "/home/jfq/pinns-tf2-main/pinns-tf2-main/examples/aneurysm3D/train.py", line 139, in main metric_dict, _ = pinnstf2.train( ^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 84, in wrap raise ex File "/root/pinns-tf2/pinnstf2/utils/utils.py", line 72, in wrap metric_dict, object_dict = task_func( ^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/train.py", line 122, in train model: PINNModule = hydra.utils.instantiate(cfg.model)( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/pinns-tf2/pinnstf2/models/pinn_module.py", line 56, in __init__ self.opt = optimizer() ^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/adam.py", line 62, in __init__ super().__init__( File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/optimizer.py", line 21, in __init__ super().__init__(*args, **kwargs) File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/optimizers/base_optimizer.py", line 158, in __init__ iterations = backend.Variable( ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/common/variables.py", line 173, in __init__ initializer = self._convert_to_tensor(initializer, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 73, in _convert_to_tensor return convert_to_tensor(value, dtype=dtype) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/keras/src/backend/tensorflow/core.py", line 152, in convert_to_tensor return tf.cast(x, dtype) ^^^^^^^^^^^^^^^^^ File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/jfq/anaconda3/envs/pinn/lib/python3.12/site-packages/tensorflow/python/framework/ops.py", line 6027, in raise_from_not_ok_status raise core._status_to_exception(e) from None # pylint: disable=protected-access ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ tensorflow.python.framework.errors_impl.InternalError: {{function_node __wrapped__Cast_device_/job:localhost/replica:0/task:0/device:GPU:0}} 'cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, 0, reinterpret_cast<CUstream>(stream), params, nullptr)' failed with 'CUDA_ERROR_INVALID_HANDLE' [Op:Cast] name: Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.
09-20
在构建深度学习模型或相关库时,若遇到 `ninja: build stopped: subcommand failed` 错误,并提示多个规则生成了相同的输出文件(如 `dcn_v2_cuda.obj`),这通常意味着构建系统检测到两个或更多不同的编译步骤试图生成同一个目标文件。这种冲突会导致构建失败。 ### 原因分析 - **源文件重复编译**:不同编译配置或路径设置可能导致同一源文件被多次编译并尝试生成相同的目标文件。 - **构建脚本配置错误**:Ninja 构建文件(`build.ninja`)中可能存在不正确的规则定义,导致多个规则指向同一输出。 - **第三方库依赖问题**:例如 DCN(Deformable Convolution)等自定义 CUDA 操作可能与项目其他部分存在依赖冲突。 ### 解决方案 - **检查源码结构和 CMakeLists.txt 配置**:确保没有重复的源文件被添加进构建系统。对于使用 CMake 的项目,应检查 `CMakeLists.txt` 文件中是否多次包含了 `dcn_v2_cuda.cpp` 或 `.cu` 文件[^1]。 - **清理构建目录并重新生成 Ninja 文件**: ```bash rm -rf build/ mkdir build && cd build cmake .. ninja ``` 这样可以避免旧的构建残留影响新构建过程。 - **修改 CUDA 架构支持参数**:如果错误信息中包含类似 `nvcc fatal : Unsupported gpu architecture ‘compute_86’`,说明当前使用的 CUDA 工具链不支持该 GPU 架构(如 RTX 3090 对应 compute_86)。可以在构建命令中显式指定支持的架构,例如在调用 `cmake` 时添加: ```bash -DCMAKE_CUDA_ARCHITECTURES=75 ``` 或者在 `setup.py` 中调整 CUDA 编译参数以排除不支持的架构[^1]。 - **更新 PyTorch 和 CUDA 环境**:某些情况下,特定版本的 PyTorch 可能与自定义 CUDA 扩展存在兼容性问题。尝试升级 PyTorch 到最新稳定版,并确认 CUDA 版本与驱动程序匹配。 - **手动编辑 `build.ninja` 文件**:作为最后手段,可打开 `build.ninja` 查找所有涉及 `dcn_v2_cuda.obj` 的规则,并删除或注释掉重复的部分。但此方法不推荐,因为下次运行 CMake 会覆盖更改[^1]。 --- ###
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值