13 lines
550 B
Bash
13 lines
550 B
Bash
|
#!/bin/bash
|
||
|
|
# Wrapper script to run Python with proper environment
|
||
|
|
|
||
|
|
# Set up library paths for CUDA
|
||
|
|
VENV_DIR="/home/koko210Serve/parakeet-test/venv/lib/python3.11/site-packages"
|
||
|
|
export LD_LIBRARY_PATH="${VENV_DIR}/nvidia/cublas/lib:${VENV_DIR}/nvidia/cudnn/lib:${VENV_DIR}/nvidia/cufft/lib:${VENV_DIR}/nvidia/cuda_nvrtc/lib:${VENV_DIR}/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH"
|
||
|
|
|
||
|
|
# Set Python path
|
||
|
|
export PYTHONPATH="/home/koko210Serve/parakeet-test:$PYTHONPATH"
|
||
|
|
|
||
|
|
# Run Python with arguments
|
||
|
|
exec /home/koko210Serve/parakeet-test/venv/bin/python "$@"
|