obsolete/version_check.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import torch
import safetensors

print(f"PyTorch version: {torch.__version__}\nsafetensors version: {safetensors.__version__}")

if torch.cuda.is_available():
    print("CUDA is available.")
else:
    print("CUDA is not available.")