文章背景图

Pytorch-Nvidia GPU安装&配置详细步骤

2026-03-01
25
- 字
- 分钟
|

本教程适用于 NVIDIA 显卡 + Windows 系统 + Conda 环境。
⚠️ 强烈建议:不要手动单独安装 CUDA Toolkit,除非你需要编译 CUDA 程序。
对于 PyTorch 用户来说,只需安装正确的 NVIDIA 驱动 + 匹配版本的 PyTorch 即可。

1. 安装Conda

Conda 可选择:

  • Anaconda(体积大,预装包多)
  • Miniconda(轻量级,推荐)

这里推荐 Miniconda。

1.1 下载Miniconda

根据操作系统和CPU架构,进入官方仓库找到对应的安装包,点击下载:Index of /miniconda

1.2 安装 Miniconda

以 Windows 为例:

  1. 右键安装包 → 以管理员身份运行
  2. 一直点击下一步
  3. 必须勾选:Add Miniconda to PATH(添加到系统路径)

1.3 验证安装

按下Win+R打开运行面板

输入powershell打开命令行

在命令行中输入:

conda --version

若输出版本号,例如:

conda 24.x.x

则说明安装成功。

1.4 创建&激活conda环境

创建环境,在powershell中输入:

conda create -n torch -y

激活环境:

conda activate torch

若在命令行前显示环境名,类似(torch) PS C:\Users\Username>,则激活成功。

1.5 为conda安装Manba

Mamba 是 Conda 的 C++ 重写版本,​完全兼容 Conda 命令​,但安装速度和依赖解析速度比 Conda 快 10-50 倍。

conda原生安装pytorch需要解析数十分钟,并且占用大量内存(32G内存能给占满)。

Mamba只需要解析几十秒,占用很少内存。

激活conda的torch环境后,在powershell中输入如下指令安装mamba:

conda install mamba -y

2 安装Pytorch

2.1 确认Nvidia驱动

安装前确认Nvidia驱动,通常在装机时,显卡驱动会自动安装

在 PowerShell 输入:

nvidia-smi

如果输出类似(注意图中的CUDA Version):

屏幕截图 2026-03-01 112415.png

说明驱动已安装。

如果提示命令不存在,则需要安装驱动(此处省略驱动安装过程)。

上图中的CUDA Version代表当前显卡驱动最高能支持的CUDA版本,记住,之后有用。

2.2 安装Pytorch

进入Pytorch官网PyTorch Versions,查找你能支持CUDA版本的Pytorch版本,找到对应的conda安装指令,例如conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia

将其中的conda替换为mamba,在激活conda环境的powershell命令行中输入:

mamba install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia

等到安装完成即可

2.3 验证安装

在激活了conda环境的命令行中输入python,回车进入

复制如下代码并粘贴:

import torch
print(torch.__version__)
print(torch.cuda.is_available())       # 应该输出 True
print(torch.cuda.get_device_name(0))   # 应该显示 你的显卡型号

示例输出:
屏幕截图 2026-03-01 112415.png

输出正常则说明大成功!

之后使用时,必须激活你为Pytorch创建的conda环境。

原创

Pytorch-Nvidia GPU安装&配置详细步骤

本文链接: Pytorch-Nvidia GPU安装&配置详细步骤

本文采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处。

评论交流

文章目录