frog_bench / frog_bench.sh
aka7774's picture
Create frog_bench.sh
d891c5e
raw
history blame contribute delete
462 Bytes
### for ROCm
git clone https://github.com/kohya-ss/sd-scripts.git
cd sd-scripts
git clone https://huggingface.co/aka7774/frog_bench.git
sudo apt install python3.10-venv
python3 -m venv venv
venv/bin/pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2
venv/bin/pip3 install --upgrade -r requirements.txt
sudo apt-get install -y libgl1-mesa-dev
venv/bin/pip3 uninstall tensorflow
venv/bin/pip3 install tensorflow-rocm