如何在Python代码中指定CUDA设备？

在Python中，如果你使用的是PyTorch库，可以利用它的`torch.cuda.set_device()`函数来指定CUDA设备。首先你需要确认设备是否存在并获取其ID，通常来说，0表示第一块GPU。以下是一个简单的例子：

import torch

# 检查当前可用的CUDA设备
device_ids = list(range(torch.cuda.device_count()))
if device_ids:
    # 设定默认设备（如果有多块GPU）
    if len(device_ids) > 1:
        print(f"Found {len(device_ids)} GPUs, setting default to GPU 0")
        torch.cuda.set_device(0)
    else:
        print(f"Using GPU: {device_ids[0]}")

    # 现在可以创建张量并在指定设备上运行
    tensor = torch.randn((10, 10), device=torch.device('cuda'))
    # 训练模型或其他计算操作会自动在选定的设备上执行
else:
    print("No CUDA devices found")

如果你使用的是TensorFlow，可以使用`tf.config.list_physical_devices('GPU')`来获取GPU设备，并设置`tf.device('/GPU:0')`等来指定。