env:
conda env create -f evomdlm.yaml
conda install -c conda-forge gcc_linux-64 gxx_linux-64
conda install -c "nvidia/label/cuda-12.4" cuda-nvcc cuda-libraries-dev
pip install flash-attn==2.7.4.post1 --no-build-isolation
# Try to import flash_attn
python -c "import flash_attn"
env:
conda env create -f evomdlm.yaml
conda install -c conda-forge gcc_linux-64 gxx_linux-64
conda install -c "nvidia/label/cuda-12.4" cuda-nvcc cuda-libraries-dev
pip install flash-attn==2.7.4.post1 --no-build-isolation
# Try to import flash_attn
python -c "import flash_attn"