pythonversions.py

import sys
import torch
import torchvision
import torchaudio

print("python version:", sys.version)
print("python version info:", sys.version_info)
print("torch version:", torch.__version__)
print("cuda version (torch):", torch.version.cuda)
print("torchvision version:", torchvision.__version__)
print("torchaudio version:", torchaudio.__version__)
print("cuda available:", torch.cuda.is_available())

try:
    import flash_attn
    print("flash-attention version:", flash_attn.__version__)
except ImportError:
    print("flash-attention is not installed or cannot be imported")

try:
    import triton
    print("triton version:", triton.__version__)
except ImportError:
    print("triton is not installed or cannot be imported")

try:
    import sageattention
    print("sageattention version:", sageattention.__version__)
except ImportError:
    print("sageattention is not installed or cannot be imported")
except AttributeError:
    print("sageattention is installed but has no __version__ attribute")

try:
    import decord
    print("decord version:", decord.__version__)
except ImportError:
    print("decord is not installed or cannot be imported")
except AttributeError:
    print("decord is installed but has no __version__ attribute")

try:
    import diffusers
    print("diffusers version:", diffusers.__version__)
except ImportError:
    print("diffusers is not installed or cannot be imported")
except AttributeError:
    print("diffusers is installed but has no __version__ attribute")

try:
    import pillow
    print("pillow version:", pillow.__version__)
except ImportError:
    print("pillow is not installed or cannot be imported")
except AttributeError:
    print("pillow is installed but has no __version__ attribute")

try:
    import einops
    print("einops version:", einops.__version__)
except ImportError:
    print("einops is not installed or cannot be imported")
except AttributeError:
    print("einops is installed but has no __version__ attribute")

try:
    import scipy
    print("scipy version:", scipy.__version__)
except ImportError:
    print("scipy is not installed or cannot be imported")
except AttributeError:
    print("scipy is installed but has no __version__ attribute")

try:
    import numpy
    print("numpy version:", numpy.__version__)
except ImportError:
    print("numpy is not installed or cannot be imported")
except AttributeError:
    print("numpy is installed but has no __version__ attribute")

try:
    import cv2
    print("opencv-python version:", cv2.__version__)
except ImportError:
    print("opencv-python is not installed or cannot be imported")
except AttributeError:
    print("opencv-python is installed but has no __version__ attribute")

try:
    import huggingface_hub
    print("huggingface_hub version:", huggingface_hub.__version__)
except ImportError:
    print("huggingface_hub is not installed or cannot be imported")
except AttributeError:
    print("huggingface_hub is installed but has no __version__ attribute")

try:
    import omegaconf
    print("omegaconf version:", omegaconf.__version__)
except ImportError:
    print("omegaconf is not installed or cannot be imported")
except AttributeError:
    print("omegaconf is installed but has no __version__ attribute")

try:
    import einops
    print("einops version:", einops.__version__)
except ImportError:
    print("einops is not installed or cannot be imported")
except AttributeError:
    print("einops is installed but has no __version__ attribute")

try:
    import accelerate
    print("accelerate version:", accelerate.__version__)
except ImportError:
    print("accelerate is not installed or cannot be imported")
except AttributeError:
    print("accelerate is installed but has no __version__ attribute")