From f28d61cb2a3969d7ad1c77c5995f613d67ab3d63 Mon Sep 17 00:00:00 2001
From: Tri Dao <tridpq@gmail.com>
Date: Mon, 13 Mar 2023 12:48:07 -0700
Subject: [PATCH] Update README on requirements (nvcc and Pytorch)

---
 README.md | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 1af5533..3969798 100644
--- a/README.md
+++ b/README.md
@@ -40,7 +40,11 @@ to what's used in our paper.
 
 ## Beta release (0.2).
 
-To install (requiring CUDA 11, NVCC, and an Turing or Ampere GPU):
+Requirements:
+- CUDA 11.4 and above.
+- PyTorch 1.12 and above.
+
+To install:
 ```sh
 pip install flash-attn
 ```