File size: 379 Bytes
6524e7a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/bash
export CUDA_VISIBLE_DEVICES=0
export CUDA_LAUNCH_BLOCKING=1
export PYTHONUNBUFFERED=1
# Check GPU availability
echo "Checking GPU availability..."
nvidia-smi
python -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}')"
# Run the application
echo "Starting NeuroNest application at $(date)"
python gradio_test.py
echo "Job completed at $(date)"
|