15 lines
543 B
Bash
15 lines
543 B
Bash
#!/bin/bash
|
|
|
|
set -e
|
|
set -o nounset
|
|
|
|
cd /tmp
|
|
# Paperspace's venv doesn't install pip for some reason
|
|
python -m venv --without-pip /tmp/lama-venv
|
|
wget https://bootstrap.pypa.io/get-pip.py
|
|
/tmp/lama-venv/bin/python get-pip.py
|
|
/tmp/lama-venv/bin/pip install torch==2.0.1+cu118 torchvision triton --extra-index-url https://download.pytorch.org/whl/cu118
|
|
/tmp/lama-venv/bin/pip install xformers==0.0.20 --no-deps
|
|
/tmp/lama-venv/bin/pip install pyre-extensions==0.0.29
|
|
/tmp/lama-venv/bin/pip install lama-cleaner
|
|
/tmp/lama-venv/bin/pip install rembg
|