setup-vm.sh

echo "Installing conda..."
# https://waylonwalker.com/install-miniconda/
mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh
~/miniconda3/bin/conda init bash

# set up pytorch
echo "Installing Python dependencies..."
source ~/miniconda3/etc/profile.d/conda.sh
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
python -m pip install --upgrade pip
pip install transformers accelerate bitsandbytes

pip freeze > requirements-from-setup.txt