解决方案：
使用镜像：docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
flash attention whl包，支持用户在Z100L&K100上使用，可直接通过本帖下方链接下载。下载后可解压到任意目录，先pip uninstall flash_attn，再通过pip install flash_attn-2.0.4_abi1_torch2.1_dtk24.04.1-cp310-cp310-linux_x86_64.whl安装即可。