Ver código fonte

Update README on requirements (nvcc and Pytorch)

Tri Dao 2 anos atrás
pai
commit
f28d61cb2a
1 arquivos alterados com 5 adições e 1 exclusões
  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
 ```