#!/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"