philipptrepte commited on
Commit
20daae6
·
verified ·
1 Parent(s): a084e14

Update BUILD.md with multi-Python build instructions

Browse files
Files changed (1) hide show
  1. BUILD.md +40 -4
BUILD.md CHANGED
@@ -2,8 +2,17 @@
2
 
3
  **Built:** 2026-04-07
4
  **Target:** DGX Spark (GB10 Blackwell, aarch64, SM 10.0 + SM 12.1)
5
- **Wheel:** `dist/tensorflow-2.21.0-cp310-cp310-linux_aarch64.whl` (377 MB)
6
- **Build time:** ~1.5 hours
 
 
 
 
 
 
 
 
 
7
 
8
  ## Build Configuration
9
 
@@ -104,19 +113,46 @@ echo 'build:cuda --repo_env LOCAL_CUDNN_PATH="/home/philipptrepte/bioai/tools/te
104
 
105
  ### 6. Build
106
 
 
 
 
107
  ```bash
 
108
  bazel build \
109
  --config=cuda_wheel \
110
  --config=opt \
111
  --copt=-march=armv9-a \
112
  --verbose_failures \
113
  --repo_env=HERMETIC_NVSHMEM_VERSION="" \
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
114
  //tensorflow/tools/pip_package:wheel
115
  ```
116
 
117
  The wheel lands at:
118
  ```
119
- bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow-2.21.0-cp310-cp310-linux_aarch64.whl
120
  ```
121
 
122
  Copy to dist:
@@ -128,7 +164,7 @@ cp bazel-bin/tensorflow/tools/pip_package/wheel_house/*.whl ../dist/
128
  ### 7. Install
129
 
130
  ```bash
131
- pip install ~/bioai/tools/tensorflow-build/dist/tensorflow-2.21.0-cp310-cp310-linux_aarch64.whl
132
  ```
133
 
134
  ### 8. Verify
 
2
 
3
  **Built:** 2026-04-07
4
  **Target:** DGX Spark (GB10 Blackwell, aarch64, SM 10.0 + SM 12.1)
5
+ **HuggingFace:** [Qanatpharma/tensorflow-2.21.0-cuda13-aarch64](https://huggingface.co/Qanatpharma/tensorflow-2.21.0-cuda13-aarch64)
6
+ **Build time:** ~2 hours per Python version
7
+
8
+ ## Available Wheels
9
+
10
+ | Python | Wheel | Size |
11
+ |--------|-------|------|
12
+ | 3.10 | `tensorflow-2.21.0-cp310-cp310-linux_aarch64.whl` | 377 MB |
13
+ | 3.11 | `tensorflow-2.21.0-cp311-cp311-linux_aarch64.whl` | 377 MB |
14
+ | 3.12 | `tensorflow-2.21.0-cp312-cp312-linux_aarch64.whl` | 377 MB |
15
+ | 3.13 | `tensorflow-2.21.0-cp313-cp313-linux_aarch64.whl` | 378 MB |
16
 
17
  ## Build Configuration
18
 
 
113
 
114
  ### 6. Build
115
 
116
+ The `HERMETIC_PYTHON_VERSION` flag controls which Python the wheel targets.
117
+ Each version requires a full rebuild (~2 hours).
118
+
119
  ```bash
120
+ # Python 3.10 (default if HERMETIC_PYTHON_VERSION is omitted)
121
  bazel build \
122
  --config=cuda_wheel \
123
  --config=opt \
124
  --copt=-march=armv9-a \
125
  --verbose_failures \
126
  --repo_env=HERMETIC_NVSHMEM_VERSION="" \
127
+ --repo_env=HERMETIC_PYTHON_VERSION=3.10 \
128
+ //tensorflow/tools/pip_package:wheel
129
+ ```
130
+
131
+ For other Python versions, change the env and re-configure first:
132
+
133
+ ```bash
134
+ # Example: Python 3.12
135
+ conda create -n tf-build-312 python=3.12 numpy -y
136
+ conda activate tf-build-312
137
+ pip install numpy wheel packaging requests
138
+
139
+ # Re-run configure (sets PYTHON_BIN_PATH for the new env)
140
+ export PYTHON_BIN_PATH=$(which python)
141
+ # ... same env vars as step 5 ...
142
+ ./configure
143
+ echo 'build:cuda --repo_env LOCAL_CUDNN_PATH=".../cudnn"' >> .tf_configure.bazelrc
144
+
145
+ # Build with matching hermetic Python
146
+ bazel build \
147
+ --config=cuda_wheel --config=opt --copt=-march=armv9-a \
148
+ --verbose_failures --repo_env=HERMETIC_NVSHMEM_VERSION="" \
149
+ --repo_env=HERMETIC_PYTHON_VERSION=3.12 \
150
  //tensorflow/tools/pip_package:wheel
151
  ```
152
 
153
  The wheel lands at:
154
  ```
155
+ bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow-2.21.0-cp3XX-cp3XX-linux_aarch64.whl
156
  ```
157
 
158
  Copy to dist:
 
164
  ### 7. Install
165
 
166
  ```bash
167
+ pip install ~/bioai/tools/tensorflow-build/dist/tensorflow-2.21.0-cp3XX-cp3XX-linux_aarch64.whl
168
  ```
169
 
170
  ### 8. Verify