Files
digital-drosophila/models/male-cns-lif/conditioning-acceleration-v1/repair_environment.py
T

23 lines
1.1 KiB
Python

"""Log repair of candidate-only packaging tool; preserve first failed attempt."""
import json
import os
from pathlib import Path
import subprocess
OUT = Path('/home/qishizhe/phase4ar-v100-acceleration-20261001')
PYTHON = OUT / '.venv/bin/python'
assert PYTHON.exists() and (OUT / 'install-candidate.stderr.log').exists()
def run(args, stem):
with (OUT / (stem + '.stdout.log')).open('x') as stdout, (OUT / (stem + '.stderr.log')).open('x') as stderr:
result = subprocess.run([str(PYTHON), '-B', '-m', 'pip', *args], stdout=stdout, stderr=stderr,
env={**os.environ, 'PYTHONDONTWRITEBYTECODE': '1'})
print(json.dumps({'step': stem, 'exit_code': result.returncode}), flush=True)
if result.returncode:
raise RuntimeError(stem)
run(['install', '--disable-pip-version-check', 'pip==25.2'], 'upgrade-candidate-pip')
run(['install', '--disable-pip-version-check', '--report', str(OUT / 'pip-install-report.json'),
'numpy==2.2.6', 'cupy-cuda12x==13.6.0'], 'install-candidate-attempt-02')
run(['freeze'], 'candidate-freeze')