Quellcode durchsuchen

Update README on requirements (nvcc and Pytorch)

Tri Dao vor 2 Jahren
Ursprung
Commit
f28d61cb2a
1 geänderte Dateien mit 5 neuen und 1 gelöschten Zeilen
  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
 ```