Explorar o código

Update README on requirements (nvcc and Pytorch)

Tri Dao %!s(int64=2) %!d(string=hai) anos
pai
achega
f28d61cb2a
Modificáronse 1 ficheiros con 5 adicións e 1 borrados
  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
 ```