Files
gogo2/start_with_gpu.sh

31 lines
773 B
Bash

#!/bin/bash
# Startup script with AMD GPU gfx1151 fix
# Set AMD GPU compatibility
export HSA_OVERRIDE_GFX_VERSION=11.0.0
# Activate virtual environment
source venv/bin/activate
# Optional: Enable experimental features for better performance
# export TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
echo "GPU Compatibility: HSA_OVERRIDE_GFX_VERSION=11.0.0"
echo "Virtual environment: $(which python)"
echo ""
echo "Starting application..."
echo ""
# Start your application (modify as needed)
# python main_dashboard.py
# or
# python ANNOTATE/web/app.py
# If you want to run a specific script, pass it as argument
if [ $# -gt 0 ]; then
python "$@"
else
echo "Usage: ./start_with_gpu.sh <your_script.py>"
echo "Example: ./start_with_gpu.sh ANNOTATE/web/app.py"
fi