DocumentQA / env_setup.sh
Epoching's picture
init
c14d9ad
raw
history blame
856 Bytes
conda create --name llnl_actici_env python=3.9
conda activate llnl_actici_env
conda install pytorch=1.10 torchvision torchaudio cudatoolkit=11.3 -c pytorch
# For DiT
python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
# For DiT
pip install opencv-python
pip install timm
pip install pdfminer.six
conda install -c conda-forge poppler
pip install pdf2image
pip install pypdf2
pip install spacy
# pytesseract, in case we need in future
pip install pytesseract
# For Retrieval & QA
pip install transformers==4.20
pip install sentence-transformers
# For Demo
pip install gradio
# If Jupyter is allowed
pip install jupyter
# (Optional, adding this custom env to the base environment's jupyter)
python -m ipykernel install --user --name llnl_actici_env --display-name "Python (llnl_actici_env)"