Add AMD GPU compatibility fix for gfx1151, including fallback to CPU mode and environment variable setup
This commit is contained in:
30
start_with_gpu.sh
Normal file
30
start_with_gpu.sh
Normal file
@@ -0,0 +1,30 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user