Update builder.sh
Browse files- builder.sh +2 -0
builder.sh
CHANGED
|
@@ -5,6 +5,8 @@ echo "🚀 Builder (FlashAttn LayerNorm extra + Apex + Q8) — runtime com GPU v
|
|
| 5 |
|
| 6 |
# ===== Config e diretórios =====
|
| 7 |
|
|
|
|
|
|
|
| 8 |
|
| 9 |
mkdir -p /app/wheels /app/cuda_cache /app/wheels/src
|
| 10 |
chmod -R 777 /app/wheels || true
|
|
|
|
| 5 |
|
| 6 |
# ===== Config e diretórios =====
|
| 7 |
|
| 8 |
+
nvcc -V
|
| 9 |
+
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
|
| 10 |
|
| 11 |
mkdir -p /app/wheels /app/cuda_cache /app/wheels/src
|
| 12 |
chmod -R 777 /app/wheels || true
|