Updated check_gpu_availability script

check_gpu_availability

@@ -2,6 +2,9 @@ import torch
 
 if torch.cuda.is_available():
     device = torch.device("cuda")
+    print("Torch version:",torch.__version__)
+    print("Is CUDA enabled?",torch.cuda.is_available())
+    print("Number of GPUs available:", torch.cuda.device_count())
     print(f"GPU {torch.cuda.get_device_name(0)} is available")
 else:
     device = torch.device("cpu")

code/check_cuda.py

@@ -1,11 +0,0 @@
-import torch
-
-print("Number of GPUs:", torch.cuda.device_count())
-
-print("Torch version:",torch.__version__)
-
-print("Is CUDA enabled?",torch.cuda.is_available())
-
-print(torch.cuda.device_count())
-
-# pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html