!pip list | grep "fastai"
fastai 2.7.17
if torch.cuda.is_available():
print("GPU is available.")
num_gpu = torch.cuda.device_count()
for i in range(num_gpu):
gpu_name = torch.cuda.get_device_name(i)
gpu_memory = torch.cuda.get_device_properties(i).total_memory / (1024 ** 3) # Convert bytes to GB
print(f"GPU {i}: {gpu_name}, Memory: {gpu_memory:.2f} GB")
else:
print("GPU is not available.")
GPU is available.
GPU 0: NVIDIA GeForce RTX 2060, Memory: 6.00 GB