You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
757 B
25 lines
757 B
#!/bin/bash
|
|
|
|
# Set CUDA environment
|
|
if [ -d "/usr/local/cuda" ]; then
|
|
export CUDA_HOME=/usr/local/cuda
|
|
echo "CUDA environment set to $CUDA_HOME"
|
|
elif [ -d "/usr/lib/cuda" ]; then
|
|
export CUDA_HOME=/usr/lib/cuda
|
|
echo "CUDA environment set to $CUDA_HOME"
|
|
else
|
|
echo "CUDA not found. Running in CPU-only mode."
|
|
fi
|
|
|
|
# Set LibTorch library path
|
|
if [ -d "$HOME/libtorch" ]; then
|
|
export LD_LIBRARY_PATH=$HOME/libtorch/lib:$LD_LIBRARY_PATH
|
|
echo "LibTorch library path set to $HOME/libtorch/lib"
|
|
elif [ -d "/usr/local/libtorch" ]; then
|
|
export LD_LIBRARY_PATH=/usr/local/libtorch/lib:$LD_LIBRARY_PATH
|
|
echo "LibTorch library path set to /usr/local/libtorch/lib"
|
|
fi
|
|
|
|
# Run the demo
|
|
echo "Starting tracking demo..."
|
|
./bin/tracking_demo
|