Skip to content

Commit db145de

Browse files
committed
fix
1 parent 5a57f43 commit db145de

File tree

1 file changed

+13
-16
lines changed

1 file changed

+13
-16
lines changed

Dockerfile

Lines changed: 13 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -77,22 +77,19 @@ RUN TORCH_CMAKE_PATH=$(python -c "import torch; print(torch.utils.cmake_prefix_p
7777
echo "CMAKE_PREFIX_PATH=$TORCH_CMAKE_PATH" >> /etc/environment
7878

7979
# 🚀 GitHub Actions优化:智能设置并行度(针对2核7GB限制)
80-
RUN python -c "
81-
import os, psutil
82-
# GitHub Actions runner: 2核心,7GB内存
83-
cpu_cores = min(2, os.cpu_count())
84-
available_memory_gb = min(7, psutil.virtual_memory().available / (1024**3))
85-
# 保守策略:每个job约3GB内存
86-
memory_jobs = max(1, int(available_memory_gb / 3))
87-
# 选择安全的并行度
88-
optimal_jobs = min(cpu_cores, memory_jobs, 2)
89-
nvcc_threads = optimal_jobs
90-
print(f'🎯 CI优化: MAX_JOBS={optimal_jobs}, NVCC_THREADS={nvcc_threads}')
91-
print(f'💾 估算资源: {available_memory_gb:.1f}GB, {cpu_cores}核')
92-
with open('/etc/environment', 'a') as f:
93-
f.write(f'MAX_JOBS={optimal_jobs}\n')
94-
f.write(f'NVCC_THREADS={nvcc_threads}\n')
95-
"
80+
RUN python -c "\
81+
import os, psutil; \
82+
cpu_cores = min(2, os.cpu_count()); \
83+
available_memory_gb = min(7, psutil.virtual_memory().available / (1024**3)); \
84+
memory_jobs = max(1, int(available_memory_gb / 3)); \
85+
optimal_jobs = min(cpu_cores, memory_jobs, 2); \
86+
nvcc_threads = optimal_jobs; \
87+
print(f'🎯 CI优化: MAX_JOBS={optimal_jobs}, NVCC_THREADS={nvcc_threads}'); \
88+
print(f'💾 估算资源: {available_memory_gb:.1f}GB, {cpu_cores}核'); \
89+
f = open('/etc/environment', 'a'); \
90+
f.write(f'MAX_JOBS={optimal_jobs}\n'); \
91+
f.write(f'NVCC_THREADS={nvcc_threads}\n'); \
92+
f.close()"
9693

9794
# Create output directory
9895
RUN mkdir -p /out

0 commit comments

Comments
 (0)