Prechádzať zdrojové kódy

Update README on requirements (nvcc and Pytorch)

Tri Dao 2 rokov pred
rodič
commit
f28d61cb2a
1 zmenil súbory, kde vykonal 5 pridanie a 1 odobranie
  1. 5 1
      README.md

+ 5 - 1
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
 ```