如何在Python中检测和确认是否使用GPU来运行代码？

在Python中检测和确认是否使用GPU通常涉及到两个库：`torch`（针对深度学习任务）和`tensorflow`（另一种常用的深度学习框架）。以下是检查这两种情况的方法：

1. **使用PyTorch**:

   import torch

   # 检查是否有可用的GPU
   device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
   print(f"当前设备: {device}")

   # 确认正在使用的设备
   if device.type == 'cuda':
       print("正在使用GPU")

2. **使用TensorFlow**:

   import tensorflow as tf

   # 检查GPU可用性
   gpus = tf.config.list_physical_devices('GPU')
   if gpus:
       print("有GPU可用")
       try:
           # 配置默认GPU使用
           tf.config.set_visible_devices(gpus[0], 'GPU')
           logical_gpus = tf.config.list_logical_devices('GPU')
           print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs")
       except RuntimeError as e:
           # Visible devices must be set before GPUs have been initialized
           print(e)
   else:
       print("无GPU可用")