Spaces:
Build error
Build error
add accelerate
Browse files- install_env.sh +2 -3
install_env.sh
CHANGED
|
@@ -1,10 +1,9 @@
|
|
| 1 |
# Run this code manually
|
| 2 |
-
# conda create -n faiss_env python=3.10
|
| 3 |
-
# conda activate faiss_env
|
| 4 |
|
| 5 |
conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.1 -c pytorch -c nvidia -y
|
| 6 |
pip install transformers datasets pyserini
|
| 7 |
|
| 8 |
conda install -c pytorch -c nvidia faiss-gpu=1.8.0 -y
|
| 9 |
|
| 10 |
-
pip install uvicorn fastapi huggingface_hub pydantic hf_transfer
|
|
|
|
| 1 |
# Run this code manually
|
| 2 |
+
# conda create -n faiss_env python=3.10 && conda activate faiss_env
|
|
|
|
| 3 |
|
| 4 |
conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.1 -c pytorch -c nvidia -y
|
| 5 |
pip install transformers datasets pyserini
|
| 6 |
|
| 7 |
conda install -c pytorch -c nvidia faiss-gpu=1.8.0 -y
|
| 8 |
|
| 9 |
+
pip install uvicorn fastapi huggingface_hub pydantic hf_transfer accelerate
|