report-review

gpu_check.py (997B)

---

 import torch
 
 # 1. CUDAが利用可能か（CUDA版PyTorchがインストールされているか）
 is_available = torch.cuda.is_available()
 print(f"torch.cuda.is_available(): {is_available}")
 
 if is_available:
     # 2. PCに認識されているGPUの数
     device_count = torch.cuda.device_count()
     print(f"torch.cuda.device_count(): {device_count}")
 
     # 3. 現在使用中のGPUの名前（通常0番目のGPU）
     device_name = torch.cuda.get_device_name(0)
     print(f"GPU Name: {device_name}")
 
     # 4. GPU上でテンソル演算を実行できるか
     try:
         # GPU (CUDA) デバイスを指定
         device = torch.device("cuda")
         # テンソルをGPUメモリに作成し、演算を実行
         tensor_on_gpu = torch.rand(2, 3, device=device)
         print("\nSuccessfully created and calculated a tensor on GPU:")
         print(tensor_on_gpu)
     except Exception as e:
         print(f"\nError accessing GPU for calculation: {e}")