computefence 0.2.4__tar.gz → 0.2.6__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- computefence-0.2.6/PKG-INFO +185 -0
- computefence-0.2.6/README.md +171 -0
- computefence-0.2.6/computefence/__init__.py +1 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/checks/storage.py +48 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/cli.py +6 -2
- computefence-0.2.6/computefence.egg-info/PKG-INFO +185 -0
- {computefence-0.2.4 → computefence-0.2.6}/pyproject.toml +1 -1
- computefence-0.2.4/PKG-INFO +0 -110
- computefence-0.2.4/README.md +0 -96
- computefence-0.2.4/computefence/__init__.py +0 -1
- computefence-0.2.4/computefence.egg-info/PKG-INFO +0 -110
- {computefence-0.2.4 → computefence-0.2.6}/LICENSE +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/checks/__init__.py +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/checks/dataset.py +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/checks/environment.py +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence/telemetry.py +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence.egg-info/SOURCES.txt +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence.egg-info/dependency_links.txt +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence.egg-info/entry_points.txt +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence.egg-info/requires.txt +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/computefence.egg-info/top_level.txt +0 -0
- {computefence-0.2.4 → computefence-0.2.6}/setup.cfg +0 -0
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: computefence
|
|
3
|
+
Version: 0.2.6
|
|
4
|
+
Summary: Don't lose another RunPod checkpoint to ephemeral disk before your job bills.
|
|
5
|
+
License-Expression: MIT
|
|
6
|
+
Requires-Python: >=3.8
|
|
7
|
+
Description-Content-Type: text/markdown
|
|
8
|
+
License-File: LICENSE
|
|
9
|
+
Requires-Dist: rich>=13.0.0
|
|
10
|
+
Requires-Dist: pandas>=1.5.0
|
|
11
|
+
Requires-Dist: typer>=0.9.0
|
|
12
|
+
Requires-Dist: requests>=2.28.0
|
|
13
|
+
Dynamic: license-file
|
|
14
|
+
|
|
15
|
+
# ComputeFence
|
|
16
|
+
|
|
17
|
+
**Don't lose another RunPod checkpoint to ephemeral disk before your job bills.**
|
|
18
|
+
|
|
19
|
+
One command before your training job starts. Catches the configuration mistakes that cost you hours of GPU compute and lost weights before billing begins.
|
|
20
|
+
|
|
21
|
+
Built for RunPod, Vast.ai, Lambda Labs, CoreWeave, Paperspace, and any bare metal GPU provider.
|
|
22
|
+
|
|
23
|
+
---
|
|
24
|
+
|
|
25
|
+
## Install
|
|
26
|
+
|
|
27
|
+
```bash
|
|
28
|
+
pip install computefence
|
|
29
|
+
```
|
|
30
|
+
|
|
31
|
+
No install required with UV:
|
|
32
|
+
|
|
33
|
+
```bash
|
|
34
|
+
uvx computefence doctor
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
---
|
|
38
|
+
|
|
39
|
+
## Usage
|
|
40
|
+
|
|
41
|
+
```bash
|
|
42
|
+
computefence doctor
|
|
43
|
+
```
|
|
44
|
+
|
|
45
|
+
With dataset validation:
|
|
46
|
+
|
|
47
|
+
```bash
|
|
48
|
+
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
49
|
+
```
|
|
50
|
+
|
|
51
|
+
With checkpoint directory validation:
|
|
52
|
+
|
|
53
|
+
```bash
|
|
54
|
+
computefence doctor --output-dir ./checkpoints
|
|
55
|
+
```
|
|
56
|
+
|
|
57
|
+
Add to your pod startup script so it runs automatically before every job:
|
|
58
|
+
|
|
59
|
+
```bash
|
|
60
|
+
pip install computefence && computefence doctor && python train.py
|
|
61
|
+
```
|
|
62
|
+
|
|
63
|
+
---
|
|
64
|
+
|
|
65
|
+
## Example Output
|
|
66
|
+
|
|
67
|
+
```
|
|
68
|
+
ComputeFence v0.2.5 — Pre-flight diagnostic
|
|
69
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
70
|
+
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
71
|
+
|
|
72
|
+
Environment
|
|
73
|
+
✓ Python 3.11.4
|
|
74
|
+
✓ PyTorch 2.1.0 detected
|
|
75
|
+
✓ CUDA available — NVIDIA A40
|
|
76
|
+
|
|
77
|
+
Storage
|
|
78
|
+
⚠ HF_HOME is not set. HuggingFace will cache to ephemeral local disk.
|
|
79
|
+
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
80
|
+
|
|
81
|
+
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB threshold)
|
|
82
|
+
Fix: Free up disk space or move checkpoints to a larger volume
|
|
83
|
+
df -h to check usage
|
|
84
|
+
|
|
85
|
+
Dataset
|
|
86
|
+
✓ No dataset path provided — skipping dataset checks
|
|
87
|
+
|
|
88
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
89
|
+
2 warning(s) found. Review before launching.
|
|
90
|
+
|
|
91
|
+
Anonymous run stats are collected to improve ComputeFence.
|
|
92
|
+
To opt out: touch ~/.computefence_no_telemetry
|
|
93
|
+
```
|
|
94
|
+
|
|
95
|
+
---
|
|
96
|
+
|
|
97
|
+
## What It Checks
|
|
98
|
+
|
|
99
|
+
**CUDA and GPU visibility**
|
|
100
|
+
Confirms PyTorch can see the GPU. Catches silent CPU fallback before it runs for eight hours at 24 seconds per iteration instead of 0.4.
|
|
101
|
+
|
|
102
|
+
**HuggingFace cache path**
|
|
103
|
+
Confirms model weights are writing to persistent storage. If HF_HOME points to ephemeral disk your weights disappear when the pod stops.
|
|
104
|
+
|
|
105
|
+
**Accelerate GPU count**
|
|
106
|
+
Confirms your distributed training config matches the GPUs actually on the instance. A mismatch leaves GPUs idle for the entire run with no error message.
|
|
107
|
+
|
|
108
|
+
**Disk space headroom**
|
|
109
|
+
Checks free space on workspace volumes and root disk. Warns below 20 GB. Blocks below 5 GB. Catches the failure where checkpoints stop saving at 90 percent completion because the disk filled.
|
|
110
|
+
|
|
111
|
+
**Checkpoint output directory**
|
|
112
|
+
Confirms your training script's output path is on persistent storage not ephemeral disk. HF_HOME and your checkpoint directory are separate paths — fixing one does not fix the other.
|
|
113
|
+
|
|
114
|
+
**Dataset integrity**
|
|
115
|
+
Optional scan for duplicates, missing values, and conflicting labels. Pass --dataset to enable.
|
|
116
|
+
|
|
117
|
+
---
|
|
118
|
+
|
|
119
|
+
## What It Does Not Check
|
|
120
|
+
|
|
121
|
+
- Training script correctness
|
|
122
|
+
- Model architecture compatibility
|
|
123
|
+
- Learning rate or hyperparameter safety
|
|
124
|
+
- Runtime behaviour during the job
|
|
125
|
+
- GPU utilisation or dataloader throughput during training
|
|
126
|
+
|
|
127
|
+
ComputeFence addresses the job configuration layer before launch. It does not monitor the run.
|
|
128
|
+
|
|
129
|
+
---
|
|
130
|
+
|
|
131
|
+
## Why This Exists
|
|
132
|
+
|
|
133
|
+
I burned approximately £1,000 on GPU training runs that failed silently.
|
|
134
|
+
|
|
135
|
+
CUDA fell back to CPU with no error message. 24 seconds per iteration instead of 0.4. Class weights caused loss to collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run finished.
|
|
136
|
+
|
|
137
|
+
Nothing existed that caught these before the job started. So I built it.
|
|
138
|
+
|
|
139
|
+
---
|
|
140
|
+
|
|
141
|
+
## Real Operator Results
|
|
142
|
+
|
|
143
|
+
**David at Neuralic** ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
144
|
+
|
|
145
|
+
**Shahzeb Ali**, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour. He would not have caught the HF_HOME issue without the tool.
|
|
146
|
+
|
|
147
|
+
**An independent Vast.ai operator** ran computefence doctor six times overnight before a real training job without being prompted or paid to do so.
|
|
148
|
+
|
|
149
|
+
Thirteen independent ML engineers confirmed this problem across RunPod, Vast.ai, and AWS. Nine had lost checkpoints to ephemeral disk. Five confirmed Docker does not solve job-specific configuration mistakes.
|
|
150
|
+
|
|
151
|
+
---
|
|
152
|
+
|
|
153
|
+
## The Problem It Solves
|
|
154
|
+
|
|
155
|
+
A healthy GPU does not mean you are running the right job.
|
|
156
|
+
|
|
157
|
+
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage that disappears on pod stop, whether your Accelerate config matches the GPUs on the instance, or whether your checkpoint output directory is on persistent storage.
|
|
158
|
+
|
|
159
|
+
ComputeFence addresses the job configuration layer — not the environment layer.
|
|
160
|
+
|
|
161
|
+
HF_HOME and your checkpoint output directory are separate paths. Fixing one does not fix the other. Both disappear on pod restart if they point to ephemeral disk.
|
|
162
|
+
|
|
163
|
+
---
|
|
164
|
+
|
|
165
|
+
## Founding Design Partner Pilot
|
|
166
|
+
|
|
167
|
+
Running high-cost GPU training jobs on RunPod or Vast.ai?
|
|
168
|
+
|
|
169
|
+
Three founding design partner spots at $99 for three months:
|
|
170
|
+
|
|
171
|
+
- Personal audit of your launch templates and persistent storage configuration
|
|
172
|
+
- ComputeFence installed into your pod startup scripts — pre-flight runs automatically before every job
|
|
173
|
+
- Slack or Discord webhook alert when a check fails or blocks a launch
|
|
174
|
+
- Monthly 30-minute call where you shape what gets built next
|
|
175
|
+
- Money back if it does not catch one actionable issue in three months
|
|
176
|
+
|
|
177
|
+
Email francisco@booth.ws to apply.
|
|
178
|
+
|
|
179
|
+
---
|
|
180
|
+
|
|
181
|
+
## Links
|
|
182
|
+
|
|
183
|
+
- GitHub: https://github.com/Francisco-Booth/ComputeFence
|
|
184
|
+
- PyPI: https://pypi.org/project/computefence
|
|
185
|
+
```
|
|
@@ -0,0 +1,171 @@
|
|
|
1
|
+
# ComputeFence
|
|
2
|
+
|
|
3
|
+
**Don't lose another RunPod checkpoint to ephemeral disk before your job bills.**
|
|
4
|
+
|
|
5
|
+
One command before your training job starts. Catches the configuration mistakes that cost you hours of GPU compute and lost weights before billing begins.
|
|
6
|
+
|
|
7
|
+
Built for RunPod, Vast.ai, Lambda Labs, CoreWeave, Paperspace, and any bare metal GPU provider.
|
|
8
|
+
|
|
9
|
+
---
|
|
10
|
+
|
|
11
|
+
## Install
|
|
12
|
+
|
|
13
|
+
```bash
|
|
14
|
+
pip install computefence
|
|
15
|
+
```
|
|
16
|
+
|
|
17
|
+
No install required with UV:
|
|
18
|
+
|
|
19
|
+
```bash
|
|
20
|
+
uvx computefence doctor
|
|
21
|
+
```
|
|
22
|
+
|
|
23
|
+
---
|
|
24
|
+
|
|
25
|
+
## Usage
|
|
26
|
+
|
|
27
|
+
```bash
|
|
28
|
+
computefence doctor
|
|
29
|
+
```
|
|
30
|
+
|
|
31
|
+
With dataset validation:
|
|
32
|
+
|
|
33
|
+
```bash
|
|
34
|
+
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
With checkpoint directory validation:
|
|
38
|
+
|
|
39
|
+
```bash
|
|
40
|
+
computefence doctor --output-dir ./checkpoints
|
|
41
|
+
```
|
|
42
|
+
|
|
43
|
+
Add to your pod startup script so it runs automatically before every job:
|
|
44
|
+
|
|
45
|
+
```bash
|
|
46
|
+
pip install computefence && computefence doctor && python train.py
|
|
47
|
+
```
|
|
48
|
+
|
|
49
|
+
---
|
|
50
|
+
|
|
51
|
+
## Example Output
|
|
52
|
+
|
|
53
|
+
```
|
|
54
|
+
ComputeFence v0.2.5 — Pre-flight diagnostic
|
|
55
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
56
|
+
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
57
|
+
|
|
58
|
+
Environment
|
|
59
|
+
✓ Python 3.11.4
|
|
60
|
+
✓ PyTorch 2.1.0 detected
|
|
61
|
+
✓ CUDA available — NVIDIA A40
|
|
62
|
+
|
|
63
|
+
Storage
|
|
64
|
+
⚠ HF_HOME is not set. HuggingFace will cache to ephemeral local disk.
|
|
65
|
+
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
66
|
+
|
|
67
|
+
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB threshold)
|
|
68
|
+
Fix: Free up disk space or move checkpoints to a larger volume
|
|
69
|
+
df -h to check usage
|
|
70
|
+
|
|
71
|
+
Dataset
|
|
72
|
+
✓ No dataset path provided — skipping dataset checks
|
|
73
|
+
|
|
74
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
75
|
+
2 warning(s) found. Review before launching.
|
|
76
|
+
|
|
77
|
+
Anonymous run stats are collected to improve ComputeFence.
|
|
78
|
+
To opt out: touch ~/.computefence_no_telemetry
|
|
79
|
+
```
|
|
80
|
+
|
|
81
|
+
---
|
|
82
|
+
|
|
83
|
+
## What It Checks
|
|
84
|
+
|
|
85
|
+
**CUDA and GPU visibility**
|
|
86
|
+
Confirms PyTorch can see the GPU. Catches silent CPU fallback before it runs for eight hours at 24 seconds per iteration instead of 0.4.
|
|
87
|
+
|
|
88
|
+
**HuggingFace cache path**
|
|
89
|
+
Confirms model weights are writing to persistent storage. If HF_HOME points to ephemeral disk your weights disappear when the pod stops.
|
|
90
|
+
|
|
91
|
+
**Accelerate GPU count**
|
|
92
|
+
Confirms your distributed training config matches the GPUs actually on the instance. A mismatch leaves GPUs idle for the entire run with no error message.
|
|
93
|
+
|
|
94
|
+
**Disk space headroom**
|
|
95
|
+
Checks free space on workspace volumes and root disk. Warns below 20 GB. Blocks below 5 GB. Catches the failure where checkpoints stop saving at 90 percent completion because the disk filled.
|
|
96
|
+
|
|
97
|
+
**Checkpoint output directory**
|
|
98
|
+
Confirms your training script's output path is on persistent storage not ephemeral disk. HF_HOME and your checkpoint directory are separate paths — fixing one does not fix the other.
|
|
99
|
+
|
|
100
|
+
**Dataset integrity**
|
|
101
|
+
Optional scan for duplicates, missing values, and conflicting labels. Pass --dataset to enable.
|
|
102
|
+
|
|
103
|
+
---
|
|
104
|
+
|
|
105
|
+
## What It Does Not Check
|
|
106
|
+
|
|
107
|
+
- Training script correctness
|
|
108
|
+
- Model architecture compatibility
|
|
109
|
+
- Learning rate or hyperparameter safety
|
|
110
|
+
- Runtime behaviour during the job
|
|
111
|
+
- GPU utilisation or dataloader throughput during training
|
|
112
|
+
|
|
113
|
+
ComputeFence addresses the job configuration layer before launch. It does not monitor the run.
|
|
114
|
+
|
|
115
|
+
---
|
|
116
|
+
|
|
117
|
+
## Why This Exists
|
|
118
|
+
|
|
119
|
+
I burned approximately £1,000 on GPU training runs that failed silently.
|
|
120
|
+
|
|
121
|
+
CUDA fell back to CPU with no error message. 24 seconds per iteration instead of 0.4. Class weights caused loss to collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run finished.
|
|
122
|
+
|
|
123
|
+
Nothing existed that caught these before the job started. So I built it.
|
|
124
|
+
|
|
125
|
+
---
|
|
126
|
+
|
|
127
|
+
## Real Operator Results
|
|
128
|
+
|
|
129
|
+
**David at Neuralic** ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
130
|
+
|
|
131
|
+
**Shahzeb Ali**, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour. He would not have caught the HF_HOME issue without the tool.
|
|
132
|
+
|
|
133
|
+
**An independent Vast.ai operator** ran computefence doctor six times overnight before a real training job without being prompted or paid to do so.
|
|
134
|
+
|
|
135
|
+
Thirteen independent ML engineers confirmed this problem across RunPod, Vast.ai, and AWS. Nine had lost checkpoints to ephemeral disk. Five confirmed Docker does not solve job-specific configuration mistakes.
|
|
136
|
+
|
|
137
|
+
---
|
|
138
|
+
|
|
139
|
+
## The Problem It Solves
|
|
140
|
+
|
|
141
|
+
A healthy GPU does not mean you are running the right job.
|
|
142
|
+
|
|
143
|
+
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage that disappears on pod stop, whether your Accelerate config matches the GPUs on the instance, or whether your checkpoint output directory is on persistent storage.
|
|
144
|
+
|
|
145
|
+
ComputeFence addresses the job configuration layer — not the environment layer.
|
|
146
|
+
|
|
147
|
+
HF_HOME and your checkpoint output directory are separate paths. Fixing one does not fix the other. Both disappear on pod restart if they point to ephemeral disk.
|
|
148
|
+
|
|
149
|
+
---
|
|
150
|
+
|
|
151
|
+
## Founding Design Partner Pilot
|
|
152
|
+
|
|
153
|
+
Running high-cost GPU training jobs on RunPod or Vast.ai?
|
|
154
|
+
|
|
155
|
+
Three founding design partner spots at $99 for three months:
|
|
156
|
+
|
|
157
|
+
- Personal audit of your launch templates and persistent storage configuration
|
|
158
|
+
- ComputeFence installed into your pod startup scripts — pre-flight runs automatically before every job
|
|
159
|
+
- Slack or Discord webhook alert when a check fails or blocks a launch
|
|
160
|
+
- Monthly 30-minute call where you shape what gets built next
|
|
161
|
+
- Money back if it does not catch one actionable issue in three months
|
|
162
|
+
|
|
163
|
+
Email francisco@booth.ws to apply.
|
|
164
|
+
|
|
165
|
+
---
|
|
166
|
+
|
|
167
|
+
## Links
|
|
168
|
+
|
|
169
|
+
- GitHub: https://github.com/Francisco-Booth/ComputeFence
|
|
170
|
+
- PyPI: https://pypi.org/project/computefence
|
|
171
|
+
```
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = "0.2.6"
|
|
@@ -134,3 +134,51 @@ def check_disk_headroom():
|
|
|
134
134
|
})
|
|
135
135
|
|
|
136
136
|
return results
|
|
137
|
+
|
|
138
|
+
|
|
139
|
+
# Paths that are wiped when a rented instance stops.
|
|
140
|
+
EPHEMERAL_PREFIXES = ["/root", "/tmp"]
|
|
141
|
+
# Paths that survive an instance restart.
|
|
142
|
+
PERSISTENT_PREFIXES = ["/workspace", "/runpod-volume", "/vast", "/home"]
|
|
143
|
+
OUTPUT_DIR_FIX = "Move your output_dir to /workspace/checkpoints or your mounted volume"
|
|
144
|
+
|
|
145
|
+
|
|
146
|
+
def _is_under(path, prefix):
|
|
147
|
+
"""True when path sits at or below prefix.
|
|
148
|
+
|
|
149
|
+
Compares path components rather than string prefixes so /workspace-old is
|
|
150
|
+
not mistaken for a child of /workspace. Avoids Path.is_relative_to, which
|
|
151
|
+
needs Python 3.9 while this package supports 3.8.
|
|
152
|
+
"""
|
|
153
|
+
path_parts = Path(path).parts
|
|
154
|
+
prefix_parts = Path(prefix).parts
|
|
155
|
+
return path_parts[: len(prefix_parts)] == prefix_parts
|
|
156
|
+
|
|
157
|
+
|
|
158
|
+
def check_output_dir(output_dir=None):
|
|
159
|
+
"""Flag a checkpoint directory that will not survive the instance stopping.
|
|
160
|
+
|
|
161
|
+
Uses abspath rather than resolve() so a path is judged as written: on macOS
|
|
162
|
+
/tmp is a symlink to /private/tmp, and resolving it first would hide the
|
|
163
|
+
very ephemeral prefix this check exists to catch.
|
|
164
|
+
"""
|
|
165
|
+
if output_dir is None:
|
|
166
|
+
return []
|
|
167
|
+
|
|
168
|
+
path = Path(os.path.abspath(os.path.expanduser(str(output_dir))))
|
|
169
|
+
|
|
170
|
+
is_ephemeral = any(_is_under(path, prefix) for prefix in EPHEMERAL_PREFIXES) or not any(
|
|
171
|
+
_is_under(path, prefix) for prefix in PERSISTENT_PREFIXES
|
|
172
|
+
)
|
|
173
|
+
|
|
174
|
+
if is_ephemeral:
|
|
175
|
+
return [{
|
|
176
|
+
"status": "fail",
|
|
177
|
+
"message": f"Output directory {path} is on ephemeral storage and will be lost when the instance stops",
|
|
178
|
+
"fix": OUTPUT_DIR_FIX
|
|
179
|
+
}]
|
|
180
|
+
|
|
181
|
+
return [{
|
|
182
|
+
"status": "pass",
|
|
183
|
+
"message": f"Output directory {path} is on persistent storage"
|
|
184
|
+
}]
|
|
@@ -2,7 +2,7 @@ import sys
|
|
|
2
2
|
from rich.console import Console
|
|
3
3
|
from computefence import __version__
|
|
4
4
|
from computefence.checks.environment import check_environment
|
|
5
|
-
from computefence.checks.storage import check_disk_headroom, check_storage
|
|
5
|
+
from computefence.checks.storage import check_disk_headroom, check_output_dir, check_storage
|
|
6
6
|
from computefence.checks.dataset import check_dataset
|
|
7
7
|
from computefence.telemetry import record_run
|
|
8
8
|
|
|
@@ -27,6 +27,7 @@ def doctor():
|
|
|
27
27
|
dataset = None
|
|
28
28
|
input_column = None
|
|
29
29
|
label_column = None
|
|
30
|
+
output_dir = None
|
|
30
31
|
args = sys.argv[1:]
|
|
31
32
|
i = 0
|
|
32
33
|
while i < len(args):
|
|
@@ -39,6 +40,9 @@ def doctor():
|
|
|
39
40
|
elif args[i] == "--label-column" and i + 1 < len(args):
|
|
40
41
|
label_column = args[i + 1]
|
|
41
42
|
i += 2
|
|
43
|
+
elif args[i] == "--output-dir" and i + 1 < len(args):
|
|
44
|
+
output_dir = args[i + 1]
|
|
45
|
+
i += 2
|
|
42
46
|
else:
|
|
43
47
|
i += 1
|
|
44
48
|
|
|
@@ -47,7 +51,7 @@ def doctor():
|
|
|
47
51
|
console.print("━" * 50)
|
|
48
52
|
|
|
49
53
|
env_results = check_environment()
|
|
50
|
-
storage_results = check_storage() + check_disk_headroom()
|
|
54
|
+
storage_results = check_storage() + check_disk_headroom() + check_output_dir(output_dir)
|
|
51
55
|
dataset_results = check_dataset(
|
|
52
56
|
dataset_path=dataset,
|
|
53
57
|
input_column=input_column,
|
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: computefence
|
|
3
|
+
Version: 0.2.6
|
|
4
|
+
Summary: Don't lose another RunPod checkpoint to ephemeral disk before your job bills.
|
|
5
|
+
License-Expression: MIT
|
|
6
|
+
Requires-Python: >=3.8
|
|
7
|
+
Description-Content-Type: text/markdown
|
|
8
|
+
License-File: LICENSE
|
|
9
|
+
Requires-Dist: rich>=13.0.0
|
|
10
|
+
Requires-Dist: pandas>=1.5.0
|
|
11
|
+
Requires-Dist: typer>=0.9.0
|
|
12
|
+
Requires-Dist: requests>=2.28.0
|
|
13
|
+
Dynamic: license-file
|
|
14
|
+
|
|
15
|
+
# ComputeFence
|
|
16
|
+
|
|
17
|
+
**Don't lose another RunPod checkpoint to ephemeral disk before your job bills.**
|
|
18
|
+
|
|
19
|
+
One command before your training job starts. Catches the configuration mistakes that cost you hours of GPU compute and lost weights before billing begins.
|
|
20
|
+
|
|
21
|
+
Built for RunPod, Vast.ai, Lambda Labs, CoreWeave, Paperspace, and any bare metal GPU provider.
|
|
22
|
+
|
|
23
|
+
---
|
|
24
|
+
|
|
25
|
+
## Install
|
|
26
|
+
|
|
27
|
+
```bash
|
|
28
|
+
pip install computefence
|
|
29
|
+
```
|
|
30
|
+
|
|
31
|
+
No install required with UV:
|
|
32
|
+
|
|
33
|
+
```bash
|
|
34
|
+
uvx computefence doctor
|
|
35
|
+
```
|
|
36
|
+
|
|
37
|
+
---
|
|
38
|
+
|
|
39
|
+
## Usage
|
|
40
|
+
|
|
41
|
+
```bash
|
|
42
|
+
computefence doctor
|
|
43
|
+
```
|
|
44
|
+
|
|
45
|
+
With dataset validation:
|
|
46
|
+
|
|
47
|
+
```bash
|
|
48
|
+
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
49
|
+
```
|
|
50
|
+
|
|
51
|
+
With checkpoint directory validation:
|
|
52
|
+
|
|
53
|
+
```bash
|
|
54
|
+
computefence doctor --output-dir ./checkpoints
|
|
55
|
+
```
|
|
56
|
+
|
|
57
|
+
Add to your pod startup script so it runs automatically before every job:
|
|
58
|
+
|
|
59
|
+
```bash
|
|
60
|
+
pip install computefence && computefence doctor && python train.py
|
|
61
|
+
```
|
|
62
|
+
|
|
63
|
+
---
|
|
64
|
+
|
|
65
|
+
## Example Output
|
|
66
|
+
|
|
67
|
+
```
|
|
68
|
+
ComputeFence v0.2.5 — Pre-flight diagnostic
|
|
69
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
70
|
+
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
71
|
+
|
|
72
|
+
Environment
|
|
73
|
+
✓ Python 3.11.4
|
|
74
|
+
✓ PyTorch 2.1.0 detected
|
|
75
|
+
✓ CUDA available — NVIDIA A40
|
|
76
|
+
|
|
77
|
+
Storage
|
|
78
|
+
⚠ HF_HOME is not set. HuggingFace will cache to ephemeral local disk.
|
|
79
|
+
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
80
|
+
|
|
81
|
+
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB threshold)
|
|
82
|
+
Fix: Free up disk space or move checkpoints to a larger volume
|
|
83
|
+
df -h to check usage
|
|
84
|
+
|
|
85
|
+
Dataset
|
|
86
|
+
✓ No dataset path provided — skipping dataset checks
|
|
87
|
+
|
|
88
|
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
89
|
+
2 warning(s) found. Review before launching.
|
|
90
|
+
|
|
91
|
+
Anonymous run stats are collected to improve ComputeFence.
|
|
92
|
+
To opt out: touch ~/.computefence_no_telemetry
|
|
93
|
+
```
|
|
94
|
+
|
|
95
|
+
---
|
|
96
|
+
|
|
97
|
+
## What It Checks
|
|
98
|
+
|
|
99
|
+
**CUDA and GPU visibility**
|
|
100
|
+
Confirms PyTorch can see the GPU. Catches silent CPU fallback before it runs for eight hours at 24 seconds per iteration instead of 0.4.
|
|
101
|
+
|
|
102
|
+
**HuggingFace cache path**
|
|
103
|
+
Confirms model weights are writing to persistent storage. If HF_HOME points to ephemeral disk your weights disappear when the pod stops.
|
|
104
|
+
|
|
105
|
+
**Accelerate GPU count**
|
|
106
|
+
Confirms your distributed training config matches the GPUs actually on the instance. A mismatch leaves GPUs idle for the entire run with no error message.
|
|
107
|
+
|
|
108
|
+
**Disk space headroom**
|
|
109
|
+
Checks free space on workspace volumes and root disk. Warns below 20 GB. Blocks below 5 GB. Catches the failure where checkpoints stop saving at 90 percent completion because the disk filled.
|
|
110
|
+
|
|
111
|
+
**Checkpoint output directory**
|
|
112
|
+
Confirms your training script's output path is on persistent storage not ephemeral disk. HF_HOME and your checkpoint directory are separate paths — fixing one does not fix the other.
|
|
113
|
+
|
|
114
|
+
**Dataset integrity**
|
|
115
|
+
Optional scan for duplicates, missing values, and conflicting labels. Pass --dataset to enable.
|
|
116
|
+
|
|
117
|
+
---
|
|
118
|
+
|
|
119
|
+
## What It Does Not Check
|
|
120
|
+
|
|
121
|
+
- Training script correctness
|
|
122
|
+
- Model architecture compatibility
|
|
123
|
+
- Learning rate or hyperparameter safety
|
|
124
|
+
- Runtime behaviour during the job
|
|
125
|
+
- GPU utilisation or dataloader throughput during training
|
|
126
|
+
|
|
127
|
+
ComputeFence addresses the job configuration layer before launch. It does not monitor the run.
|
|
128
|
+
|
|
129
|
+
---
|
|
130
|
+
|
|
131
|
+
## Why This Exists
|
|
132
|
+
|
|
133
|
+
I burned approximately £1,000 on GPU training runs that failed silently.
|
|
134
|
+
|
|
135
|
+
CUDA fell back to CPU with no error message. 24 seconds per iteration instead of 0.4. Class weights caused loss to collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run finished.
|
|
136
|
+
|
|
137
|
+
Nothing existed that caught these before the job started. So I built it.
|
|
138
|
+
|
|
139
|
+
---
|
|
140
|
+
|
|
141
|
+
## Real Operator Results
|
|
142
|
+
|
|
143
|
+
**David at Neuralic** ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
144
|
+
|
|
145
|
+
**Shahzeb Ali**, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour. He would not have caught the HF_HOME issue without the tool.
|
|
146
|
+
|
|
147
|
+
**An independent Vast.ai operator** ran computefence doctor six times overnight before a real training job without being prompted or paid to do so.
|
|
148
|
+
|
|
149
|
+
Thirteen independent ML engineers confirmed this problem across RunPod, Vast.ai, and AWS. Nine had lost checkpoints to ephemeral disk. Five confirmed Docker does not solve job-specific configuration mistakes.
|
|
150
|
+
|
|
151
|
+
---
|
|
152
|
+
|
|
153
|
+
## The Problem It Solves
|
|
154
|
+
|
|
155
|
+
A healthy GPU does not mean you are running the right job.
|
|
156
|
+
|
|
157
|
+
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage that disappears on pod stop, whether your Accelerate config matches the GPUs on the instance, or whether your checkpoint output directory is on persistent storage.
|
|
158
|
+
|
|
159
|
+
ComputeFence addresses the job configuration layer — not the environment layer.
|
|
160
|
+
|
|
161
|
+
HF_HOME and your checkpoint output directory are separate paths. Fixing one does not fix the other. Both disappear on pod restart if they point to ephemeral disk.
|
|
162
|
+
|
|
163
|
+
---
|
|
164
|
+
|
|
165
|
+
## Founding Design Partner Pilot
|
|
166
|
+
|
|
167
|
+
Running high-cost GPU training jobs on RunPod or Vast.ai?
|
|
168
|
+
|
|
169
|
+
Three founding design partner spots at $99 for three months:
|
|
170
|
+
|
|
171
|
+
- Personal audit of your launch templates and persistent storage configuration
|
|
172
|
+
- ComputeFence installed into your pod startup scripts — pre-flight runs automatically before every job
|
|
173
|
+
- Slack or Discord webhook alert when a check fails or blocks a launch
|
|
174
|
+
- Monthly 30-minute call where you shape what gets built next
|
|
175
|
+
- Money back if it does not catch one actionable issue in three months
|
|
176
|
+
|
|
177
|
+
Email francisco@booth.ws to apply.
|
|
178
|
+
|
|
179
|
+
---
|
|
180
|
+
|
|
181
|
+
## Links
|
|
182
|
+
|
|
183
|
+
- GitHub: https://github.com/Francisco-Booth/ComputeFence
|
|
184
|
+
- PyPI: https://pypi.org/project/computefence
|
|
185
|
+
```
|
|
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
|
|
|
5
5
|
[project]
|
|
6
6
|
name = "computefence"
|
|
7
7
|
dynamic = ["version"]
|
|
8
|
-
description = "
|
|
8
|
+
description = "Don't lose another RunPod checkpoint to ephemeral disk before your job bills."
|
|
9
9
|
readme = "README.md"
|
|
10
10
|
requires-python = ">=3.8"
|
|
11
11
|
license = "MIT"
|
computefence-0.2.4/PKG-INFO
DELETED
|
@@ -1,110 +0,0 @@
|
|
|
1
|
-
Metadata-Version: 2.4
|
|
2
|
-
Name: computefence
|
|
3
|
-
Version: 0.2.4
|
|
4
|
-
Summary: Pre-flight validation for GPU training runs on rented infrastructure
|
|
5
|
-
License-Expression: MIT
|
|
6
|
-
Requires-Python: >=3.8
|
|
7
|
-
Description-Content-Type: text/markdown
|
|
8
|
-
License-File: LICENSE
|
|
9
|
-
Requires-Dist: rich>=13.0.0
|
|
10
|
-
Requires-Dist: pandas>=1.5.0
|
|
11
|
-
Requires-Dist: typer>=0.9.0
|
|
12
|
-
Requires-Dist: requests>=2.28.0
|
|
13
|
-
Dynamic: license-file
|
|
14
|
-
|
|
15
|
-
# ComputeFence
|
|
16
|
-
|
|
17
|
-
Pre-flight safety gate for ML training jobs on rented GPU compute. Built for RunPod, Vast.ai, Lambda Labs, and similar bare metal providers.
|
|
18
|
-
|
|
19
|
-
## Install
|
|
20
|
-
|
|
21
|
-
```bash
|
|
22
|
-
pip install computefence
|
|
23
|
-
```
|
|
24
|
-
|
|
25
|
-
Or with UV:
|
|
26
|
-
|
|
27
|
-
```bash
|
|
28
|
-
uvx computefence doctor
|
|
29
|
-
```
|
|
30
|
-
|
|
31
|
-
## Usage
|
|
32
|
-
|
|
33
|
-
```bash
|
|
34
|
-
computefence doctor
|
|
35
|
-
```
|
|
36
|
-
|
|
37
|
-
With a dataset:
|
|
38
|
-
|
|
39
|
-
```bash
|
|
40
|
-
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
41
|
-
```
|
|
42
|
-
|
|
43
|
-
## Example output
|
|
44
|
-
|
|
45
|
-
ComputeFence v0.2.0 — Pre-flight diagnostic
|
|
46
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
47
|
-
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
48
|
-
|
|
49
|
-
Environment
|
|
50
|
-
✓ Python 3.11.4
|
|
51
|
-
✓ PyTorch 2.1.0 detected
|
|
52
|
-
✓ CUDA available — NVIDIA A40
|
|
53
|
-
|
|
54
|
-
Storage
|
|
55
|
-
⚠ HF_HOME is not set. HuggingFace will use default local cache.
|
|
56
|
-
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
57
|
-
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB)
|
|
58
|
-
Fix: Free up disk space or attach a larger volume before launching
|
|
59
|
-
|
|
60
|
-
Dataset
|
|
61
|
-
✓ No dataset path provided — skipping dataset checks
|
|
62
|
-
|
|
63
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
64
|
-
2 warning(s) found. Review before launching.
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
## What it checks
|
|
68
|
-
|
|
69
|
-
- **CUDA and GPU visibility** — confirms PyTorch can see the GPU and training will not silently fall back to CPU
|
|
70
|
-
- **HuggingFace cache path** — confirms model files go to persistent storage not ephemeral disk
|
|
71
|
-
- **Accelerate GPU count** — confirms your distributed training config matches the GPUs actually on the instance
|
|
72
|
-
- **Disk space headroom** — checks free space on workspace volumes and root disk. Warns below 20 GB, blocks below 5 GB
|
|
73
|
-
- **Dataset integrity** — optional scan for duplicates, missing values, and conflicting labels
|
|
74
|
-
|
|
75
|
-
## What it does not check
|
|
76
|
-
|
|
77
|
-
- Training script correctness
|
|
78
|
-
- Model architecture compatibility
|
|
79
|
-
- Learning rate or hyperparameter safety
|
|
80
|
-
- Runtime monitoring during the job
|
|
81
|
-
- Slow dataloader or data pipeline throughput
|
|
82
|
-
- Dataloader bottleneck causing low GPU utilisation
|
|
83
|
-
|
|
84
|
-
## Why this exists
|
|
85
|
-
|
|
86
|
-
I burned approximately £1,000 on GPU training runs that failed silently. CUDA fell back to CPU with no error. 24 seconds per iteration instead of 0.4. Class weights caused loss collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run.
|
|
87
|
-
|
|
88
|
-
Nothing existed that caught these before the job started. So I built it.
|
|
89
|
-
|
|
90
|
-
## Real operator results
|
|
91
|
-
|
|
92
|
-
David at Neuralic ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
93
|
-
|
|
94
|
-
Shahzeb Ali, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour and would not have caught the HF_HOME issue explicitly without the tool.
|
|
95
|
-
|
|
96
|
-
## The problem it solves
|
|
97
|
-
|
|
98
|
-
A healthy GPU does not mean you are training the right job.
|
|
99
|
-
|
|
100
|
-
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage, whether your Accelerate config matches the GPUs actually on the instance, or whether your disk has enough headroom for checkpoints. ComputeFence addresses the job layer not the environment layer.
|
|
101
|
-
|
|
102
|
-
Five independent operators confirmed this independently. Docker does not solve job-specific configuration mistakes.
|
|
103
|
-
|
|
104
|
-
## GitHub
|
|
105
|
-
|
|
106
|
-
github.com/Francisco-Booth/ComputeFence
|
|
107
|
-
|
|
108
|
-
## PyPI
|
|
109
|
-
|
|
110
|
-
pypi.org/project/computefence
|
computefence-0.2.4/README.md
DELETED
|
@@ -1,96 +0,0 @@
|
|
|
1
|
-
# ComputeFence
|
|
2
|
-
|
|
3
|
-
Pre-flight safety gate for ML training jobs on rented GPU compute. Built for RunPod, Vast.ai, Lambda Labs, and similar bare metal providers.
|
|
4
|
-
|
|
5
|
-
## Install
|
|
6
|
-
|
|
7
|
-
```bash
|
|
8
|
-
pip install computefence
|
|
9
|
-
```
|
|
10
|
-
|
|
11
|
-
Or with UV:
|
|
12
|
-
|
|
13
|
-
```bash
|
|
14
|
-
uvx computefence doctor
|
|
15
|
-
```
|
|
16
|
-
|
|
17
|
-
## Usage
|
|
18
|
-
|
|
19
|
-
```bash
|
|
20
|
-
computefence doctor
|
|
21
|
-
```
|
|
22
|
-
|
|
23
|
-
With a dataset:
|
|
24
|
-
|
|
25
|
-
```bash
|
|
26
|
-
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
27
|
-
```
|
|
28
|
-
|
|
29
|
-
## Example output
|
|
30
|
-
|
|
31
|
-
ComputeFence v0.2.0 — Pre-flight diagnostic
|
|
32
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
33
|
-
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
34
|
-
|
|
35
|
-
Environment
|
|
36
|
-
✓ Python 3.11.4
|
|
37
|
-
✓ PyTorch 2.1.0 detected
|
|
38
|
-
✓ CUDA available — NVIDIA A40
|
|
39
|
-
|
|
40
|
-
Storage
|
|
41
|
-
⚠ HF_HOME is not set. HuggingFace will use default local cache.
|
|
42
|
-
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
43
|
-
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB)
|
|
44
|
-
Fix: Free up disk space or attach a larger volume before launching
|
|
45
|
-
|
|
46
|
-
Dataset
|
|
47
|
-
✓ No dataset path provided — skipping dataset checks
|
|
48
|
-
|
|
49
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
50
|
-
2 warning(s) found. Review before launching.
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
## What it checks
|
|
54
|
-
|
|
55
|
-
- **CUDA and GPU visibility** — confirms PyTorch can see the GPU and training will not silently fall back to CPU
|
|
56
|
-
- **HuggingFace cache path** — confirms model files go to persistent storage not ephemeral disk
|
|
57
|
-
- **Accelerate GPU count** — confirms your distributed training config matches the GPUs actually on the instance
|
|
58
|
-
- **Disk space headroom** — checks free space on workspace volumes and root disk. Warns below 20 GB, blocks below 5 GB
|
|
59
|
-
- **Dataset integrity** — optional scan for duplicates, missing values, and conflicting labels
|
|
60
|
-
|
|
61
|
-
## What it does not check
|
|
62
|
-
|
|
63
|
-
- Training script correctness
|
|
64
|
-
- Model architecture compatibility
|
|
65
|
-
- Learning rate or hyperparameter safety
|
|
66
|
-
- Runtime monitoring during the job
|
|
67
|
-
- Slow dataloader or data pipeline throughput
|
|
68
|
-
- Dataloader bottleneck causing low GPU utilisation
|
|
69
|
-
|
|
70
|
-
## Why this exists
|
|
71
|
-
|
|
72
|
-
I burned approximately £1,000 on GPU training runs that failed silently. CUDA fell back to CPU with no error. 24 seconds per iteration instead of 0.4. Class weights caused loss collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run.
|
|
73
|
-
|
|
74
|
-
Nothing existed that caught these before the job started. So I built it.
|
|
75
|
-
|
|
76
|
-
## Real operator results
|
|
77
|
-
|
|
78
|
-
David at Neuralic ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
79
|
-
|
|
80
|
-
Shahzeb Ali, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour and would not have caught the HF_HOME issue explicitly without the tool.
|
|
81
|
-
|
|
82
|
-
## The problem it solves
|
|
83
|
-
|
|
84
|
-
A healthy GPU does not mean you are training the right job.
|
|
85
|
-
|
|
86
|
-
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage, whether your Accelerate config matches the GPUs actually on the instance, or whether your disk has enough headroom for checkpoints. ComputeFence addresses the job layer not the environment layer.
|
|
87
|
-
|
|
88
|
-
Five independent operators confirmed this independently. Docker does not solve job-specific configuration mistakes.
|
|
89
|
-
|
|
90
|
-
## GitHub
|
|
91
|
-
|
|
92
|
-
github.com/Francisco-Booth/ComputeFence
|
|
93
|
-
|
|
94
|
-
## PyPI
|
|
95
|
-
|
|
96
|
-
pypi.org/project/computefence
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
__version__ = "0.2.4"
|
|
@@ -1,110 +0,0 @@
|
|
|
1
|
-
Metadata-Version: 2.4
|
|
2
|
-
Name: computefence
|
|
3
|
-
Version: 0.2.4
|
|
4
|
-
Summary: Pre-flight validation for GPU training runs on rented infrastructure
|
|
5
|
-
License-Expression: MIT
|
|
6
|
-
Requires-Python: >=3.8
|
|
7
|
-
Description-Content-Type: text/markdown
|
|
8
|
-
License-File: LICENSE
|
|
9
|
-
Requires-Dist: rich>=13.0.0
|
|
10
|
-
Requires-Dist: pandas>=1.5.0
|
|
11
|
-
Requires-Dist: typer>=0.9.0
|
|
12
|
-
Requires-Dist: requests>=2.28.0
|
|
13
|
-
Dynamic: license-file
|
|
14
|
-
|
|
15
|
-
# ComputeFence
|
|
16
|
-
|
|
17
|
-
Pre-flight safety gate for ML training jobs on rented GPU compute. Built for RunPod, Vast.ai, Lambda Labs, and similar bare metal providers.
|
|
18
|
-
|
|
19
|
-
## Install
|
|
20
|
-
|
|
21
|
-
```bash
|
|
22
|
-
pip install computefence
|
|
23
|
-
```
|
|
24
|
-
|
|
25
|
-
Or with UV:
|
|
26
|
-
|
|
27
|
-
```bash
|
|
28
|
-
uvx computefence doctor
|
|
29
|
-
```
|
|
30
|
-
|
|
31
|
-
## Usage
|
|
32
|
-
|
|
33
|
-
```bash
|
|
34
|
-
computefence doctor
|
|
35
|
-
```
|
|
36
|
-
|
|
37
|
-
With a dataset:
|
|
38
|
-
|
|
39
|
-
```bash
|
|
40
|
-
computefence doctor --dataset train.csv --input-column text --label-column label
|
|
41
|
-
```
|
|
42
|
-
|
|
43
|
-
## Example output
|
|
44
|
-
|
|
45
|
-
ComputeFence v0.2.0 — Pre-flight diagnostic
|
|
46
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
47
|
-
2 WARNINGS · 0 BLOCKERS · 3 PASSED
|
|
48
|
-
|
|
49
|
-
Environment
|
|
50
|
-
✓ Python 3.11.4
|
|
51
|
-
✓ PyTorch 2.1.0 detected
|
|
52
|
-
✓ CUDA available — NVIDIA A40
|
|
53
|
-
|
|
54
|
-
Storage
|
|
55
|
-
⚠ HF_HOME is not set. HuggingFace will use default local cache.
|
|
56
|
-
Fix: export HF_HOME=/workspace/.cache/huggingface
|
|
57
|
-
⚠ Root disk (/) — 14.3 GB free of 460.4 GB (below 20 GB)
|
|
58
|
-
Fix: Free up disk space or attach a larger volume before launching
|
|
59
|
-
|
|
60
|
-
Dataset
|
|
61
|
-
✓ No dataset path provided — skipping dataset checks
|
|
62
|
-
|
|
63
|
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
|
64
|
-
2 warning(s) found. Review before launching.
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
## What it checks
|
|
68
|
-
|
|
69
|
-
- **CUDA and GPU visibility** — confirms PyTorch can see the GPU and training will not silently fall back to CPU
|
|
70
|
-
- **HuggingFace cache path** — confirms model files go to persistent storage not ephemeral disk
|
|
71
|
-
- **Accelerate GPU count** — confirms your distributed training config matches the GPUs actually on the instance
|
|
72
|
-
- **Disk space headroom** — checks free space on workspace volumes and root disk. Warns below 20 GB, blocks below 5 GB
|
|
73
|
-
- **Dataset integrity** — optional scan for duplicates, missing values, and conflicting labels
|
|
74
|
-
|
|
75
|
-
## What it does not check
|
|
76
|
-
|
|
77
|
-
- Training script correctness
|
|
78
|
-
- Model architecture compatibility
|
|
79
|
-
- Learning rate or hyperparameter safety
|
|
80
|
-
- Runtime monitoring during the job
|
|
81
|
-
- Slow dataloader or data pipeline throughput
|
|
82
|
-
- Dataloader bottleneck causing low GPU utilisation
|
|
83
|
-
|
|
84
|
-
## Why this exists
|
|
85
|
-
|
|
86
|
-
I burned approximately £1,000 on GPU training runs that failed silently. CUDA fell back to CPU with no error. 24 seconds per iteration instead of 0.4. Class weights caused loss collapse to 0.693 immediately. My dataset had 28,432 duplicate rows and 312 conflicting labels I only found after the run.
|
|
87
|
-
|
|
88
|
-
Nothing existed that caught these before the job started. So I built it.
|
|
89
|
-
|
|
90
|
-
## Real operator results
|
|
91
|
-
|
|
92
|
-
David at Neuralic ran ComputeFence on a RunPod A100. It caught HF_HOME writing to /root/.cache and an Accelerate GPU count mismatch. He fixed both before launch.
|
|
93
|
-
|
|
94
|
-
Shahzeb Ali, a computer vision engineer running client training jobs on RunPod, confirmed the storage warning matches real pod behaviour and would not have caught the HF_HOME issue explicitly without the tool.
|
|
95
|
-
|
|
96
|
-
## The problem it solves
|
|
97
|
-
|
|
98
|
-
A healthy GPU does not mean you are training the right job.
|
|
99
|
-
|
|
100
|
-
Docker makes environments reproducible. It does not check whether your HuggingFace cache is writing to ephemeral storage, whether your Accelerate config matches the GPUs actually on the instance, or whether your disk has enough headroom for checkpoints. ComputeFence addresses the job layer not the environment layer.
|
|
101
|
-
|
|
102
|
-
Five independent operators confirmed this independently. Docker does not solve job-specific configuration mistakes.
|
|
103
|
-
|
|
104
|
-
## GitHub
|
|
105
|
-
|
|
106
|
-
github.com/Francisco-Booth/ComputeFence
|
|
107
|
-
|
|
108
|
-
## PyPI
|
|
109
|
-
|
|
110
|
-
pypi.org/project/computefence
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|