Update Dockerfile
Browse files- 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 |
-
|
|
|
|
|
|
|
| 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 \
|