安装cuda版本为12.1的pytorch

安装PyTorch时指定CUDA版本为12.1，需要按照以下步骤操作，假设你的操作系统是Ubuntu或者Debian，并且已经安装了CUDA toolkit和其他必要的依赖：

1. **更新包列表**:

   sudo apt-get update

2. **安装CUDA dependencies**:

   sudo apt-get install -y libncurses5-dev libfontconfig1-dev libxmu6 libxi6 libglu1-mesa-dev libegl1-mesa-dev

3. **下载CUDA Toolkit for Ubuntu**:
- 访问NVIDIA官网：https://developer.nvidia.com/cuda-downloads
- 搜索并下载适用于你的Linux内核的CUDA 12.1版本的.deb文件。例如，如果你的内核是 focal (20.04)，选择对应的版本下载。

4. **安装CUDA**:

   sudo dpkg -i cuda-12.1.*_linux_x64.deb

请替换`*`为实际下载的.deb文件名。

5. **安装cuDNN**:
NVIDA通常会提供cuDNN的deb包，同样从NVIDIA官网下载对应版本并安装。

6. **添加CUDA路径到环境变量**:

   echo "export PATH=/usr/local/cuda-12.1/bin:$PATH" >> ~/.bashrc
   source ~/.bashrc

7. **安装PyTorch with CUDA**:
使用以下命令安装特定版本的PyTorch，它将自动检测和链接到安装的CUDA库：

   pip install torch torchvision cudatoolkit=12.1 -f https://download.pytorch.org/whl/torch_stable.html

8. **验证安装**:

   import torch
   print(torch.cuda.get_device_name(0))  # 应该显示为"NVIDIA GPU 0"
   print(torch.cuda.is_available())      # 如果安装成功，这将打印True

记得替换上述命令中的具体版本和文件名，以匹配你的下载内容。