diff --git a/install.sh b/install.sh
index eea46d3e4..935d8cd08 100644
--- a/install.sh
+++ b/install.sh
@@ -3,6 +3,6 @@
 pip install -e .
 
 gpu_state="$(nvidia-smi --query-gpu=name --format=csv,noheader)"
-if [[ "${gpu_state}" == *"A100"* || "${gpu_state}" == *"A40"* ]]; then
+if [[ "${gpu_state}" == *"A100"* || "${gpu_state}" == *"A40"* || "${gpu_state}" == *"A6000"* ]]; then
   pip install flash-attn==2.0.2
-fi
\ No newline at end of file
+fi