Reorganize project layout and consolidate model assets
This commit is contained in:
parent
8ff11d2863
commit
b6a00b4438
15 changed files with 24 additions and 980 deletions
|
|
@ -1,24 +0,0 @@
|
|||
#!/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 "$@"
|
||||
Loading…
Add table
Add a link
Reference in a new issue