【pytorch】新的windows电脑从头搭建pytorch深度学习环境（完整版+附安装包）

文章目录

• 新的windows电脑搭建pytorch深度学习环境
• • 电脑环境的配置
  • • 显卡驱动
    • cuda
    • cudnn
    • pytorch开发软件的安装
    • • miniconda
      • vscode
      • pytorch环境的安装
      • • conda安装python环境
        • 安装pytorch和torchvision
        • 附录1：部分torch、torchvision、torchaudio版本对应关系
        • 附录2：本文涉及的软件的网盘链接

          新的windows电脑搭建pytorch深度学习环境

          最近新买了一台台式机，配置是i5 13600KF + 3060Ti。我本身在工作中台式机上是使用pytorch上进行深度学习，因此需要从头搭建pytorch环境。此文算是记录一下，如果有人需要参考也是可以的。

          电脑环境的配置

          新电脑首先需要对环境进行一定的配置，简而言之就是装一下驱动+软件。

          包括显卡驱动、CUDA、cudnn。以下是具体流程。

          显卡驱动

          一般显卡组装完成电脑是会自动安装驱动的，打开命令行敲nvidia-smi

          就可以看到显卡驱动以及对应的版本。如下图，我装好的驱动是535.98，可以支持到的CUDA版本是12.2，因此我安装的CUDA只需要在12.2以下就可以了。

          

          如果想要安装其他版本的驱动可以去英伟达驱动官网去下载。

          https://www.nvidia.cn/geforce/drivers/

          打开后是如下界面，根据自己显卡的型号就可以选择相应的驱动了。

          

          下面是搜索结果，包括应用于Game或者Studio的驱动，可以是对游戏或者视频剪辑啥的有特殊优化？这个我不太了解，我选择的是Studio的驱动。点进去下载就行了。

          

          cuda

          cuda安装是为了深度学习的GPU加速运算平台。cuda版本的安装需要根据上面显卡驱动来安装。可以到下面英伟达官网去下载

          https://developer.nvidia.com/cuda-toolkit-archive

          界面如下，选择