我最近在升级原来的环境(硬件是K100_AI,软件环境欧拉22.03)。主要目的是想要安装vllm-0.9.2+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl。
为此,我首先升级了驱动到rock-6.3.30-V1.4.1a.run。然后更新了DTK到DTK-26.04-openEuler22.03-x86_64.tar.gz。
接着安装了torch-2.5.1+das.opt1.dtk2604-cp311-cp311-manylinux_2_28_x86_64.whl;triton-3.1.0+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl。
但是安装vllm-0.9.2+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl的过程中,提示需要flash_attn2.6.1,自动安装报错No module named 'torch'。
手动进入python,运行import torch无报错。
退而求其次,手动安装了flash_attn-2.8.3+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl。
再次安装vllm-0.9.2+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl还是一样报错。
使用pip install vllm-0.9.2+das.opt1.dtk2604.torch251-cp311-cp311-manylinux_2_28_x86_64.whl -c constraints.txt安装,强制要求flash_attn的版本为2.8.3或者flash_attn==2.8.3+das.opt1.dtk2604.torch251。 依然会报错The conflict is caused by:
vllm 0.9.2+das.opt1.dtk2604.torch251 depends on flash_attn==2.6.1
The user requested (constraint) flash-attn==2.8.3+das.opt1.dtk2604.torch251
这种情况下,该如何解决,是否可以安装旧版flash\_attn-2.6.1+das.opt1.dtk25042-cp311-cp311-manylinux\_2\_28\_x86\_64.whl解决?