Skip to content

INSTALLATION.md

Latest commit

 

History

History
53 lines (38 loc) · 1.34 KB

INSTALLATION.md

File metadata and controls

53 lines (38 loc) · 1.34 KB

🛠️ Installation

  • Clone this repository:

    git clone https://github.com/EmbodiedGPT/EmbodiedGPT_Pytorch

  • Create a conda virtual environment and activate it:

    conda create -n robohusky python=3.9.16 -y
conda activate robohusky

  • Install PyTorch>=2.0 and torchvision>=0.15.2 with CUDA>=11.7:

    For examples, to install torch==2.0.1 with CUDA==11.8:

    conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
# or
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

  • Install flash-attn:

    git clone https://github.com/Dao-AILab/flash-attention.git
cd flash-attention
pip install flash-attn --no-build-isolation

  • Install transformers==4.34.1:

    pip install transformers==4.34.1

  • Install apex (optional):

    git clone https://github.com/NVIDIA/apex.git
git checkout 2386a912164b0c5cfcd8be7a2b890fbac5607c82  # https://github.com/NVIDIA/apex/issues/1735
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./

  • Install other requirements:

    cd ..
pip install -e EmbodiedGPT_Pytorch