Update BUILD.md with multi-Python build instructions
Browse files
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 |
-
**
|
| 6 |
-
**Build time:** ~
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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-
|
| 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-
|
| 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
|