cuda10.2按照pytorch

CUDA (Compute Unified Device Architecture) 是一种由 NVIDIA 开发的并行计算平台和编程模型，用于加速深度学习等高性能计算任务。CUDA 10.2 版本是在 PyTorch 中使用 GPU 进行计算的一个重要版本。

在 PyTorch 10.2 及以上版本中，你可以按照以下步骤安装和配置 CUDA：

1. **下载 CUDA 和 cuDNN**: 首先，访问 NVIDIA Developer Zone 下载对应你的操作系统和 GPU 的 CUDA SDK 和 cuDNN库。cuDNN 提供了深度学习加速的支持。

2. **安装 CUDA**:
- 完成下载后，按照安装向导安装 CUDA。
- 安装过程中通常会自动检测你的系统设置，并选择合适的路径。

3. **添加环境变量**: 将 CUDA 和 cuDNN 的路径添加到系统的 PATH 环境变量中，以便 Python 能够识别。

4. **安装 PyTorch with CUDA**:
- 使用 pip 或 conda 安装 torch torchvision 等包时，指定 cudatoolkit 版本，例如：

     pip install torch==1.8.0+cu102 torchvision==0.9.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

这里 `cu102` 表示 CUDA 10.2 版本。

5. **检查安装**:
- 在 Python 中导入 torch 并打印设备信息确认安装成功：

     import torch
     print(torch.cuda.is_available())  # 如果有可用的GPU，返回 True
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")