attention check

import sys
try:
     import torch, torchvision, torchaudio
except ImportError:
     print("Warning: Torch/Vision/Audio not fully installed yet.")

print("Python:", sys.version)
if 'torch' in sys.modules:
     print("Torch:", torch.__version__)
     print("CUDA available:", torch.cuda.is_available())
     print("Torch CUDA version:", torch.version.cuda)

try:
     import triton
     print("Triton:", triton.__version__)
except:
     print("Triton: Not installed")

try:
     import flash_attn
     print("Flash-Attn: Installed")
except:
     print("Flash-Attn: Not installed")

try:
     import sageattention
     print("SageAttention: Installed")
except:
     print("SageAttention: Not installed")