unilm/kosmos-2/docs/install.md

Setup

1. Clone the repo:

git clone https://github.com/microsoft/unilm.git
cd unilm/kosmos-2

2. Create conda env and install packages:

conda create -n kosmos-2 python=3.9
conda activate kosmos-2
pip3 install -r requirements.txt

3. Install NVIDIA/apex

git clone https://github.com/NVIDIA/apex
cd apex
# if pip >= 23.1 (ref: https://pip.pypa.io/en/stable/news/#v23-1) which supports multiple `--config-settings` with the same key... 
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" ./
# otherwise
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --global-option="--cpp_ext" --global-option="--cuda_ext" ./

4. Install the packages:

bash vl_setup_xl.sh