DERBACH/latent-rubric / scripts /environment.py
DERBACH's picture
download
raw
912 Bytes
import datetime, json, subprocess, platform
from pathlib import Path
import torch
out=['# Environment','',datetime.datetime.now(datetime.timezone.utc).isoformat(),'','Dedicated physical device: GPU1 / GPU-2a9c799b-ab9e-c677-db0e-379f480cb00f. Other GPUs are out of scope.','',f'Python: {platform.python_version()}; PyTorch: {torch.__version__}; PyTorch CUDA: {torch.version.cuda}.','', 'Environment: .venv, based on read-only system packages from /opt/anaconda3/envs/drkernel; local overrides installed inside .venv. Default system Python torch import failed (missing libcupti.so.12).']
for cmd in [['nvidia-smi'],['nvidia-smi','--query-gpu=index,uuid,name,memory.total,memory.free,driver_version','--format=csv'],['lscpu'],['free','-h'],['df','-h','.']]:
out+=['','## '+' '.join(cmd),'','```',subprocess.check_output(cmd,text=True).strip(),'```']
Path('reports/environment.md').write_text('\n'.join(out)+'\n')

Xet Storage Details

Size:
912 Bytes
·
Xet hash:
354f86c907cd20fb014db1612788b5d05a216d7632747d7d51f201cbf1b8d4e8

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.