cuda12.6 的pytorch安装

CUDA 12.6 和 PyTorch 的安装通常涉及到以下几个步骤：

1. **检查系统兼容性**：首先，确认你的系统支持 CUDA。CUDA 12.6 主要是针对 NVIDIA GPU，所以你需要有 NVIDIA 的显卡，并且确保你的操作系统（如 Windows、macOS 或 Linux）与 CUDA 版本兼容。

2. **下载 CUDA SDK**：访问 NVIDIA Developer Portal 下载 CUDA 12.6，然后选择适合你系统的版本，包括对应的操作系统、Driver 和 Toolkit。

3. **安装 CUDA**：按照安装向导的提示操作，一般会涉及添加环境变量、安装CUDA Toolkit 部分，以及驱动程序更新。

4. **安装 cuDNN**：cuDNN 是用于加速深度学习计算的库，也是PyTorch的一部分。从NVIDIA Developer Portal 下载cuDNN，确保它与你的CUDA版本相匹配。

5. **安装 PyTorch**：
- 使用 conda：```conda install pytorch torchvision cudatoolkit=12.6 -c pytorch```
- 或者 pip：```pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu126/torch_stable.html```

6. **验证安装**：安装完成后，通过运行 `torch.cuda.is_available()` 检查CUDA是否成功加载，`torch.backends.cudnn.version()` 可查看cuDNN版本。