开发验证

python3 - <<'PY'
import flash_attn 
import torch  

print("torch_cuda=" + str(torch.version.cuda)) 
print("flash_attn_version=" + flash_attn.__version__) 
print("cuda_built=" + str(torch.backends.cuda.is_built())) 
print("cuda_available=" + str(torch.cuda.is_available())) 
assert torch.backends.cuda.is_built() 
assert torch.cuda.is_available() 
assert flash_attn.__version__ == "2.8.3"
PY

torch_cuda=13.0 
flash_attn_version=2.8.3 
cuda_built=True 
cuda_available=True