Skip to content

GitLab

Commit f28d61cb authored by Tri Dao's avatar Tri Dao
Browse files

Update README on requirements (nvcc and Pytorch)

parent 57ee6181
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
README.md
View file @ f28d61cb
......@@ -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
```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment