yrshi commited on
Commit
3dfa5e3
·
1 Parent(s): 0fe2213

add accelerate

Browse files
Files changed (1) hide show
  1. 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