| set -euo pipefail | |
| conda create -n data_mining python=3.10 -y | |
| eval "$(conda shell.bash hook)" | |
| conda activate data_mining | |
| python -m pip install --upgrade pip | |
| python -m pip install -r requirements.txt | |
| python -m pip install facenet-pytorch==2.6.0 --no-deps | |
| python - <<'PY' | |
| import torch | |
| print("torch", torch.__version__) | |
| print("cuda available", torch.cuda.is_available()) | |
| if torch.cuda.is_available(): | |
| print("gpu", torch.cuda.get_device_name(0)) | |
| x = torch.randn(1024, 1024, device="cuda") | |
| y = x @ x | |
| torch.cuda.synchronize() | |
| print("cuda smoke test", float(y[0, 0])) | |
| PY | |