ViuAI commited on
Commit
3104bbb
Β·
verified Β·
1 Parent(s): 7385121

Set USE_COMPILE=0 in RTX 5090 runner to bypass Triton gcc dependency

Browse files
runners/RTX_5090_BLACKWELL_PRETRAIN.txt CHANGED
@@ -1,26 +1,28 @@
1
  # ==============================================================================
2
  # πŸš€ ViuAI Sarus-500M β€” NVIDIA RTX 5090 Blackwell Turbo Pretraining (32GB VRAM)
3
  # ==============================================================================
4
- # Instructions for JupyterLab on RTX 5090 (Blackwell sm_120):
 
5
  #
6
- # πŸ“Œ CELL 1: Run once to install CUDA 12.8 PyTorch with native sm_120 Blackwell support.
7
- # ------------------------------------------------------------------------------
8
- # !pip install --force-reinstall --no-deps -q torch --index-url https://download.pytorch.org/whl/cu128
9
- # !pip install -q transformers datasets tokenizers huggingface_hub numpy hf_transfer
10
- #
11
- # πŸ“Œ CELL 2: Run to launch continuation pretraining to 60,000 steps!
12
- # ------------------------------------------------------------------------------
13
 
14
  import os, sys, shutil
15
  import torch
16
  from huggingface_hub import HfApi, hf_hub_download
17
 
18
- # Real-time unbuffered logging, Rust-accelerated downloads & Blackwell compatibility
19
  os.environ["PYTHONUNBUFFERED"] = "1"
20
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
21
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
22
  os.environ["TORCH_CUDA_ARCH_LIST"] = "12.0;9.0+PTX"
23
 
 
 
 
24
  HF_TOKEN = os.environ.get("HF_TOKEN") or ''.join(['hf_','ssyCVhuny','XxjGdqKp','VLPpkmWK','FrrMOIFbg'])
25
  os.environ["HF_TOKEN"] = HF_TOKEN
26
 
 
1
  # ==============================================================================
2
  # πŸš€ ViuAI Sarus-500M β€” NVIDIA RTX 5090 Blackwell Turbo Pretraining (32GB VRAM)
3
  # ==============================================================================
4
+ # Instructions:
5
+ # Open JupyterLab on your RTX 5090 pod, paste this cell, and run (Shift + Enter).
6
  #
7
+ # βœ… Blackwell sm_120 Ready: Runs natively via PyTorch cu128 + Native BF16 SDPA
8
+ # βœ… No C Compiler Needed: USE_COMPILE=0 avoids Triton gcc dependencies
9
+ # βœ… Auto-resumes from Step 42,750 (already on disk, no re-downloading!)
10
+ # βœ… Streams 200 shards (10B tokens) from ViuAI/viuai-500m-data
11
+ # ==============================================================================
 
 
12
 
13
  import os, sys, shutil
14
  import torch
15
  from huggingface_hub import HfApi, hf_hub_download
16
 
17
+ # Real-time unbuffered logging & Blackwell configuration
18
  os.environ["PYTHONUNBUFFERED"] = "1"
19
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
20
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
21
  os.environ["TORCH_CUDA_ARCH_LIST"] = "12.0;9.0+PTX"
22
 
23
+ # Disable torch.compile to bypass missing C compiler (gcc) β€” uses native high-speed BF16 SDPA
24
+ os.environ["USE_COMPILE"] = "0"
25
+
26
  HF_TOKEN = os.environ.get("HF_TOKEN") or ''.join(['hf_','ssyCVhuny','XxjGdqKp','VLPpkmWK','FrrMOIFbg'])
27
  os.environ["HF_TOKEN"] = HF_TOKEN
28