File size: 773 Bytes
7fa69d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | #!/bin/bash
set -x
pip install nvidia-cusparselt-cu12==0.6.2 --no-deps \
-i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com \
2>&1 | tail -5
echo ""
echo "=== Full Verification ==="
python3 << 'PYEOF'
import torch
print(f"torch: {torch.__version__}")
print(f"CUDA: {torch.version.cuda}")
print(f"CXX11_ABI: {torch._C._GLIBCXX_USE_CXX11_ABI}")
try:
from flash_attn import flash_attn_func
import flash_attn
print(f"flash_attn: {flash_attn.__version__} - OK")
except Exception as e:
print(f"flash_attn ERROR: {e}")
import vllm
print(f"vLLM: {vllm.__version__}")
import transformers
print(f"transformers: {transformers.__version__}")
import verl
print("veRL: OK")
print("ALL_VERIFIED_OK")
PYEOF
echo "CUSPARSELT_DONE"
|