检测有几个gpu  然后使用 import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"指定gpu即可

device = torch.device('cuda:1' if torch.cuda.is_available() else 'cpu')
print('3',device)

这段代码的作用是设置环境变量 `CUDA_VISIBLE_DEVICES`，指定程序在运行时只使用特定的 GPU（在这里是编号为 1 的 GPU）。这种设置通常用于在多 GPU 环境下控制程序使用哪个 GPU 进行计算，特别是在需要避免 GPU 资源冲突或者指定 GPU 资源的情况下。

### 代码解释与用途

```python
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
```

- **`import os`**：导入 Python 的 os 模块，提供了与操作系统交互的功能。
- **`os.environ["CUDA_VISIBLE_DEVICES"] = "1"`**：设置环境变量 `CUDA_VISIBLE_DEVICES` 为 "1"，指定只使用编号为 1 的 GPU。GPU 编号从 0 开始计数。

### 使用场景

1. **多 GPU 环境下的资源管理**：
   - 在多 GPU 环境中，指定某个 GPU 运行程序，以避免与其他程序冲突或资源竞争。例如，假设机器有 4 个 GPU（编号从 0 到 3），这段代码会使得程序仅在 GPU 1 上运行。

2. **调试与开发**：
   - 在调试或开发阶段，开发者可能只想使用特定的 GPU，以便更好地控制和监控资源的使用。

3. **避免 GPU 资源冲突**：
   - 如果多个用户或程序在同一台服务器上运行，指定 GPU 可以避免多个程序争用同一个 GPU 资源，导致计算性能下降或程序崩溃。

### 示例应用

假设你有一个深度学习训练脚本 `train.py`，在多 GPU 服务器上运行时，你希望只使用 GPU 1 进行训练，可以在脚本的开头添加上述代码：

```python
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

import torch
import torch.nn as nn
import torch.optim as optim

# 其他训练代码
```

### 注意事项

- **GPU 编号从 0 开始**：如果机器有多个 GPU，它们的编号是从 0 开始的。比如，编号为 1 表示第二个 GPU。
- **设置前要检查 GPU 可用性**：确保编号为 1 的 GPU 是可用且未被其他重要任务占用的，可以使用 `nvidia-smi` 命令查看 GPU 使用情况。
- **环境变量设置的时机**：这段代码应该在导入任何使用 GPU 的库（如 PyTorch、TensorFlow）之前执行，以确保这些库在初始化时只看到指定的 GPU。

### 总结

通过设置 `CUDA_VISIBLE_DEVICES` 环境变量，您可以在多 GPU 环境中灵活地管理 GPU 资源，指定程序使用哪个 GPU，从而优化资源利用率和程序性能。这种方法在深度学习模型训练、资源调度和多用户环境中非常实用。