YUNTA88 commited on
Commit
af23746
·
verified ·
1 Parent(s): 17a3ad6

Upload root_scripts/verify_env.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. root_scripts/verify_env.sh +27 -0
root_scripts/verify_env.sh ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ echo "=== Version Check ==="
3
+ python3 << 'PYEOF'
4
+ import torch
5
+ print(f"PyTorch: {torch.__version__}")
6
+ print(f"CUDA: {torch.version.cuda}")
7
+ print(f"CXX11_ABI: {torch._C._GLIBCXX_USE_CXX11_ABI}")
8
+
9
+ import vllm
10
+ print(f"vLLM: {vllm.__version__}")
11
+
12
+ import flash_attn
13
+ print(f"flash_attn: {flash_attn.__version__}")
14
+
15
+ import transformers
16
+ print(f"transformers: {transformers.__version__}")
17
+
18
+ import verl
19
+ print("veRL: OK")
20
+
21
+ # Test flash_attn actually works
22
+ from flash_attn import flash_attn_func
23
+ print("flash_attn_func: OK")
24
+
25
+ print("ALL_CHECKS_PASSED")
26
+ PYEOF
27
+