安装支持cuda12.6的pytorch

安装支持CUDA 12.6的PyTorch通常需要几个步骤，首先确保你的系统满足以下条件：

1. **操作系统**：Ubuntu 18.04 或更高版本，Windows或macOS都行，但是官方推荐Linux环境。

2. **Python**：安装Python 3.7或以上版本。

3. **CUDA**：下载并安装CUDA 12.6，可以从NVIDIA官网下载对应版本的CUDA Toolkit。

4. **cuDNN**： CUDA深度学习库，随CUDA一起安装或者单独从NVIDIA官网下载cuDNN对应版本。

5. **Anaconda或Miniconda**（可选，用于创建独立的虚拟环境）：为了管理依赖，你可以选择安装Anaconda或Miniconda。

以下是安装步骤：

- 更新系统包列表：

   sudo apt-get update

- 安装CUDA：
- 根据CUDA下载链接下载适合你系统的.deb文件，然后运行`dpkg -i cuda-12.6.*_linux.run`
- 配置环境变量和完成安装过程。

- 安装cuDNN：
- 下载cuDNN对应版本，解压后将`libcudnn.so.8`复制到`/usr/local/cuda/lib64`目录下，并更新LD_LIBRARY_PATH环境变量。

- 安装PyTorch：

   conda create -n pytorch_cuda12.6 python=3.7 torchvision cudatoolkit=12.6 -c pytorch

或者

   pip install torch==1.9.0+cu126 torchvision cudatoolkit=12.6 -f https://download.pytorch.org/whl/torch_stable.html

- 激活新环境（如果你使用了conda）：

   conda activate pytorch_cuda12.6

- 测试安装：

   python -c "import torch; print(torch.cuda.is_available())"

如果遇到问题，可能需要查阅NVIDIA的文档或搜索解决方案。