2023-02-18 02:46:44 +00:00
|
|
|
#!/bin/bash
|
2023-02-17 00:08:27 +00:00
|
|
|
python -m venv venv
|
|
|
|
|
source ./venv/bin/activate
|
|
|
|
|
python -m pip install --upgrade pip
|
|
|
|
|
# ROCM
|
|
|
|
|
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 # 5.2 does not work for me desu
|
|
|
|
|
python -m pip install -r ./requirements.txt
|
2023-02-17 15:53:50 +00:00
|
|
|
./setup-training.sh
|
2023-02-17 00:08:27 +00:00
|
|
|
deactivate
|
