Harshit Ghosh commited on
Commit
b005e33
·
1 Parent(s): e319a61

Worker of pytorch limitation

Browse files
Files changed (1) hide show
  1. start.sh +3 -1
start.sh CHANGED
@@ -1,5 +1,7 @@
1
  #!/bin/bash
2
-
 
 
3
  # Start Celery worker in background
4
  # We use concurrency=2 to avoid memory overload on the 16GB free tier
5
  celery -A tasks worker --loglevel=info --concurrency=2 -O fair -B &
 
1
  #!/bin/bash
2
+ # Limit PyTorch background threads so Gunicorn isn't starved
3
+ export OMP_NUM_THREADS=1
4
+ export MKL_NUM_THREADS=1
5
  # Start Celery worker in background
6
  # We use concurrency=2 to avoid memory overload on the 16GB free tier
7
  celery -A tasks worker --loglevel=info --concurrency=2 -O fair -B &