rikunarita-2 commited on
Commit
02f78cf
·
verified ·
1 Parent(s): 3f52be4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -60,7 +60,7 @@ RUN wget -nv -c https://huggingface.co/circlestone-labs/Anima/resolve/main/split
60
  RUN mkdir temp && chmod 777 temp
61
 
62
  # ------------------------------------------------
63
- # 5. 起動コマンド(動的検知 + 確実なコア割り当て
64
  # ------------------------------------------------
65
  CMD quota_v2="/sys/fs/cgroup/cpu.max"; \
66
  quota_v1="/sys/fs/cgroup/cpu/cpu.cfs_quota_us"; \
@@ -82,9 +82,12 @@ CMD quota_v2="/sys/fs/cgroup/cpu.max"; \
82
  fi; \
83
  export OMP_NUM_THREADS=$COMPUTE_CORES && \
84
  export MKL_NUM_THREADS=$COMPUTE_CORES && \
 
85
  export OMP_PROC_BIND=CLOSE && \
86
  export OMP_PLACES=CORES && \
87
- echo "=== 🚀 Detected Allocated CPU Cores: $COMPUTE_CORES ===" && \
 
 
88
  python main.py \
89
  --cpu \
90
  --listen 0.0.0.0 \
 
60
  RUN mkdir temp && chmod 777 temp
61
 
62
  # ------------------------------------------------
63
+ # 5. 起動コマンド(スレッド緊縛・CPU全力駆
64
  # ------------------------------------------------
65
  CMD quota_v2="/sys/fs/cgroup/cpu.max"; \
66
  quota_v1="/sys/fs/cgroup/cpu/cpu.cfs_quota_us"; \
 
82
  fi; \
83
  export OMP_NUM_THREADS=$COMPUTE_CORES && \
84
  export MKL_NUM_THREADS=$COMPUTE_CORES && \
85
+ export TORCH_NUM_THREADS=$COMPUTE_CORES && \
86
  export OMP_PROC_BIND=CLOSE && \
87
  export OMP_PLACES=CORES && \
88
+ export OMP_SCHEDULE=STATIC && \
89
+ export KMP_AFFINITY=granularity=fine,compact,1,0 && \
90
+ echo "=== 🚀 Max Dynamic CPU Threading Engaged: $COMPUTE_CORES ===" && \
91
  python main.py \
92
  --cpu \
93
  --listen 0.0.0.0 \