squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README on requirements (nvcc and Pytorch)

Browse Source
This commit is contained in:
Tri Dao 2023-03-13 12:48:07 -07:00
parent 57ee618170
commit f28d61cb2a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -40,7 +40,11 @@ to what's used in our paper.
## Beta release (0.2). ## 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 ```sh
pip install flash-attn pip install flash-attn
``` ```