gpu_test.py

import torch

print(f"is availeble = {torch.cuda.is_available()}")

print(f"device count = {torch.cuda.device_count()}")

print(f"current device = {torch.cuda.current_device()}")