环境依赖问题：DLL load failed while importing flash_attn_2_cuda

[junglehu]

如题，其他环境都安装上了，flash_attn组件也尝试重新安装，但是这个flash_attn是没解决。

python 3.10
cude：118
GPU：3080Ti

Logs
Traceback (most recent call last):
File "E:\AIGC\ComfyUI_windows\ComfyUI\nodes.py", line 1800, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in call_with_frames_removed
File "E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA_init.py", line 1, in
from .nodes import NODE_CLASS_MAPPINGS
File "E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA\nodes.py", line 12, in
from .DragNUWA_net import Net, args
File "E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA\DragNUWA_net.py", line 5, in
from .dragnuwa.svd.modules.diffusionmodules.video_model_flow import VideoUNet_flow, VideoResBlock_Embed
File "E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA\dragnuwa\svd\modules_init_.py", line 1, in
from .encoders.modules import GeneralConditioner
File "E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA\dragnuwa\svd\modules\encoders\modules.py", line 6, in
import kornia
File "E:\Python\Python310\lib\site-packages\kornia_init_.py", line 8, in
from . import augmentation, color, contrib, core, enhance, feature, io, losses, metrics, morphology, tracking, utils, x
File "E:\Python\Python310\lib\site-packages\kornia\augmentation_init_.py", line 2, in
from kornia.augmentation import auto, container
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto_init_.py", line 1, in
from .autoaugment import AutoAugment
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto\autoaugment_init_.py", line 1, in
from .autoaugment import AutoAugment
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto\autoaugment\autoaugment.py", line 5, in
from kornia.augmentation.auto.base import SUBPLOLICY_CONFIG, PolicyAugmentBase
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto\base.py", line 5, in
from kornia.augmentation.auto.operations.base import OperationBase
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto\operations_init_.py", line 3, in
from .policy import PolicySequential
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\auto\operations\policy.py", line 7, in
from kornia.augmentation.container.base import ImageSequentialBase, TransformMatrixMinIn
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\container_init_.py", line 1, in
from kornia.augmentation.container.augment import AugmentationSequential
File "E:\Python\Python310\lib\site-packages\kornia\augmentation\container\augment.py", line 4, in
from kornia.augmentation.2d.base import RigidAffineAugmentationBase2D
File "E:\Python\Python310\lib\site-packages\kornia\augmentation_2d_init.py", line 2, in
from kornia.augmentation.2d.intensity import *
File "E:\Python\Python310\lib\site-packages\kornia\augmentation_2d\intensity_init.py", line 21, in
from kornia.augmentation.2d.intensity.plasma import RandomPlasmaBrightness, RandomPlasmaContrast, RandomPlasmaShadow
File "E:\Python\Python310\lib\site-packages\kornia\augmentation_2d\intensity\plasma.py", line 5, in
from kornia.contrib import diamond_square
File "E:\Python\Python310\lib\site-packages\kornia\contrib_init.py", line 15, in
from .image_stitching import ImageStitcher
File "E:\Python\Python310\lib\site-packages\kornia\contrib\image_stitching.py", line 7, in
from kornia.feature import LocalFeatureMatcher, LoFTR
File "E:\Python\Python310\lib\site-packages\kornia\feature_init_.py", line 6, in
from .integrated import (
File "E:\Python\Python310\lib\site-packages\kornia\feature\integrated.py", line 16, in
from .lightglue import LightGlue
File "E:\Python\Python310\lib\site-packages\kornia\feature\lightglue.py", line 16, in
from flash_attn.modules.mha import FlashCrossAttention
File "E:\Python\Python310\lib\site-packages\flash_attn_init_.py", line 3, in
from flash_attn.flash_attn_interface import (
File "E:\Python\Python310\lib\site-packages\flash_attn\flash_attn_interface.py", line 10, in
import flash_attn_2_cuda as flash_attn_cuda
ImportError: DLL load failed while importing flash_attn_2_cuda: 找不到指定的模块。

Cannot import E:\AIGC\ComfyUI_windows\ComfyUI\custom_nodes\ComfyUI-DragNUWA module for custom nodes: DLL load failed while importing flash_attn_2_cuda: 找不到指定的模块。

[Suiji12]

请问解决了吗？