@@ -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
9895RUN mkdir -p /out
0 commit comments