File size: 580 Bytes
6d35aff | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | #!/usr/bin/env bash
set -euo pipefail
repo_root="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
ape_root="${repo_root}/approach/ovod/APE"
python -B -c 'import torch; print("PyTorch", torch.__version__, "CUDA", torch.version.cuda)'
test -f "${ape_root}/setup.py"
# Dependencies must already be installed in the selected CUDA environment.
# FORCE_CUDA keeps the build explicit even on a login node without an active GPU.
FORCE_CUDA="${FORCE_CUDA:-1}" \
python -B -m pip install --no-build-isolation -e "${ape_root}"
cd "${repo_root}"
python -B scripts/check_environment.py
|