# MAKE SURE THESE TWO COMMANDS WORK FIRST
# sudo /anaconda/bin/conda init bash
# conda create --name addition
# conda activate addition

sudo apt-get update
sudo apt-get install -y python3-pip

conda update conda
conda update anaconda

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch -y
conda install -y -c conda-forge tensorboard

# Install transformers from source to get PEFT
pip3 install trl[peft]
pip3 install bitsandbytes loralib
pip3 install git+https://github.com/huggingface/transformers.git@main

pip3 install IPython
pip3 install matplotlib
pip3 install pandas
pip3 install accelerate
pip3 install sentencepiece
pip3 install bitsandbytes
pip3 install wandb
pip3 install datasets
pip3 install gitpython
pip3 install pytorch-lightning
pip3 install deepspeed
pip3 install line_profiler[ipython]
pip3 install scipy

mkdir logs

git config --global core.editor "vim"
