以下の requirements.txt を作成する。
requirements.txt
torch --index-url https://download.pytorch.org/whl/cu118
torchvision --index-url https://download.pytorch.org/whl/cu118
transformers
accelerate
sentencepiece
einops
xformers
protobuf==3.20.*
bitsandbytes==0.41.1; platform_system != "Windows"
https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.1-py3-none-win_amd64.whl; platform_system == "Windows"
peft
datasets
さらに以下のバッチファイルを作る。
install.bat
python -m venv venv
call venv\Scripts\activate
pip install -r requirements.txt
install.batを実行する。
シェルスクリプトなら多分こう。
install.sh
#!/bin/bash
python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt