13 lines
349 B
Bash
Executable File
13 lines
349 B
Bash
Executable File
#!/bin/bash
|
|
cd "$(dirname "$0")" || exit
|
|
python3 -m venv venv
|
|
source venv/bin/activate
|
|
|
|
# CPU-only PyTorch (no CUDA dependency)
|
|
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
# Remaining deps from Tsinghua mirror
|
|
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
mkdir -p checkpoints data
|