import torch
# Check the PyTorch version
print("Torch Version:", torch.__version__)
# Check if PyTorch has CUDA support enabled (should be True)
print("CUDA Available:", torch.cuda.is_available())
# Print the CUDA version PyTorch was compiled with
print("CUDA Version used by Torch:", torch.version.cuda)