<reponame>AkibMashrur/Research
import torch

def main():
    device = 'cuda' if torch.cuda.is_available() else 'cpu'
    print(f"Device available: {device}")

if __name__ == "__main__":
    main()
    