Skip to content

GitLab

"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "c7855b3b9dd05c8263735b4a706096923cbb7509"
Commit f28d61cb authored by Tri Dao's avatar Tri Dao
Browse files

Update README on requirements (nvcc and Pytorch)

parent 57ee6181
Hide 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. ...@@ -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
``` ```
......
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