25 lines
486 B
Bash
25 lines
486 B
Bash
|
#!/usr/bin/env bash
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
cd "$(dirname "$0")"
|
||
|
|
|
||
|
|
source .venv/bin/activate
|
||
|
|
|
||
|
|
CUDA_LIBS="$(python - <<'PY'
|
||
|
|
import site, os, glob
|
||
|
|
dirs=[]
|
||
|
|
for p in site.getsitepackages():
|
||
|
|
dirs += glob.glob(os.path.join(p,"nvidia","*","lib"))
|
||
|
|
dirs=[d for d in dirs if os.path.isdir(d)]
|
||
|
|
seen=set(); out=[]
|
||
|
|
for d in dirs:
|
||
|
|
if d not in seen:
|
||
|
|
seen.add(d); out.append(d)
|
||
|
|
print(":".join(out))
|
||
|
|
PY
|
||
|
|
)"
|
||
|
|
|
||
|
|
export LD_LIBRARY_PATH="${CUDA_LIBS}:${LD_LIBRARY_PATH:-}"
|
||
|
|
|
||
|
|
exec python infer.py "$@"
|