cuda12.1对应的pytorch下载

CUDA 12.1是NVIDIA针对PyTorch等深度学习框架提供的一种GPU计算平台，它允许使用Python编写的应用程序加速其计算密集型任务。为了在Python中安装PyTorch对应CUDA 12.1，你需要按照以下步骤操作：

1. **检查你的系统兼容性**：首先确保你的系统支持CUDA 12.1和对应的操作系统（Windows, Linux, 或 macOS）。

2. **安装CUDA**：访问NVIDIA Developer网站，下载CUDA 12.1的对应版本。下载并按照指示安装，记得在安装过程中选择将CUDA添加到系统的PATH环境变量中。

3. **安装cuDNN**：cuDNN是CUDA的一个重要组成部分，用于加速深度学习中的卷积神经网络。从NVIDIA Developer网站下载cuDNN，并将其与CUDA安装在同一目录下。

4. **安装PyTorch**：打开命令行或终端，选择一个Python版本（推荐使用PyTorch支持的版本），然后使用以下命令安装PyTorch。注意要指定`--cuda-version=12.1`参数：

   pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html --index-url https://download.pytorch.org/whl/nightly/cu121.html

5. **验证安装**：安装完成后，你可以通过以下代码测试是否成功安装了PyTorch与CUDA 12.1：

   import torch
   print(torch.cuda.is_available())  # 如果输出True，表示安装成功