精确安装PyTorch

精确安装PyTorch

在这篇文章中，我们将带你快速搭建 PyTorch 环境，重点讲授怎样根据你的显卡和项目需求选择符合的 CUDA Toolkit 版本，以及怎样安装符合项目要求的 PyTorch及其扩展库。
1. 环境预备

在开始安装之前，我们须要相识你的硬件与体系环境，为后续安装做足预备。
安装流程图：

1.1 确认显卡型号与驱动支持

  

• 目标：确定你的 NVIDIA GPU 型号，并确认当前驱动支持的 CUDA 版本范围。  
  
• 使用发起：
  
     
  
  • Windows 与 Linux 用户都可以使用 nvidia-smi 下令获取详细信息。如下图：
    -
    
    
    第一步必须记着cuda版本号，这里是12.6（反面要用到版本号匹配下载对应的版本）   
    
    
  

1.2 查抄驱动兼容性与 CUDA 版本需求

  

• 目标：根据你的 GPU 型号和当前驱动，确定保举使用的 CUDA Toolkit 版本。  
  
• 使用发起：
  
     
  
  • 访问 NVIDIA CUDA GPUs 列表 查询你的显卡 Compute Capability；   
    
  • 确认驱动步调支持的 CUDA 版本，从而为后续选择做参考。
    首页的通常是最新版本，大概不兼容，要与你的cuda版本匹配，如下图：
    
    
    
    进入选择列表，与上一步的cuda版本要对应。
    
    
    
    选择精确的版本