CUDA相关的模型该怎么适配解决
收藏回复举报
CUDA相关的模型该怎么适配解决
t('forum.solved') 已解决
发表于2024-11-26 16:55:04
0 查看

from torch.utils.cpp_extension import CUDA_HOME

from torch.utils.cpp_extension import CppExtension

from torch.utils.cpp_extension import CUDAExtension

模型中有用到这些,迁移到npu后,这部分代码会报错,该怎么把cuda的模型迁移到npu上后解决这部分问题,以及cuda算子该怎么适配

image.png

我要发帖子