#!/bin/sh
venv=/opt/viska/venv
cuda_libs=
for cuda_lib in "$venv"/lib/python*/site-packages/nvidia/cublas/lib \
                "$venv"/lib/python*/site-packages/nvidia/cudnn/lib; do
  if [ -d "$cuda_lib" ]; then
    cuda_libs="${cuda_libs:+$cuda_libs:}$cuda_lib"
  fi
done
if [ -n "$cuda_libs" ]; then
  export LD_LIBRARY_PATH="$cuda_libs${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
fi
exec /opt/viska/venv/bin/viska "$@"
