深度学习环境搭建:CUDA+PyTorch+TorchVision+Torchaudio 一站式安装教程

深度学习入门的第一步,从来不是学习模型原理,而是搭建一个稳定可用的运行环境------其中CUDA、PyTorch、TorchVision、Torchaudio更是核心基础,直接决定后续模型训练、代码运行能否顺畅进行。很多新手入门时,常会被"版本不兼容""安装失败""GPU调用异常"等问题卡住,耗费大量时间排查。今天这篇实操指南,就手把手带大家从零开始,一步步完成CUDA的安装、验证,以及PyTorch+TorchVision+Torchaudio的配套安装,全程避开常见坑,哪怕是零基础新手,也能轻松搞定,快速开启深度学习之路。

一、安装CUDA

CUDA 是英伟达(NVIDIA)推出的、让普通程序员能利用英伟达显卡(GPU)做高速并行计算的工具集(包含编程模型、编译器、库等),核心是把 GPU 从 "单纯玩游戏 / 渲染" 的硬件,变成能跑深度学习、科学计算的 "超级计算器",也是 PyTorch/TensorFlow 等框架能实现 GPU 加速的基础

1.首先在命令提示符中查看电脑支持的最高版本

输入命令查看:

bash 复制代码
nvidia-smi

2.下载CUDA安装包

官网:

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

进入官网界面,下滑出现所有版本

注意:尽量选择低于电脑版本,以防版本过高的不稳定的情况

我这里选择11.8版本,选择操作系统、版本等

安装包下载完成,双击打开

点击同意

选择自定义,下一步

出现此界面将CUDA展开将Visual Studio Integration取消选中,选中Visual Studio Integration可能会导致出现问题,点击下一步

出现安装位置,新手最好不要修改安装位置,点击下一步

出现这一步就安装成功,点击下一步,结束安装

3.验证CUDA是否安装成功

检查环境变量

在设置中找到系统

找到高级系统设置

点击环境变量,在系统变量查看有无CUDA的环境变量,若没有要手动添加

或者在C盘输入路径中输入cmd进入命令提示符

在命令提示符中输入以下命令查看

bash 复制代码
set cuda

输入nvcc -V验证CUDA是否安装成功

出现图片中的红色方框出来的语句就证明安装成功,如果没有说明版本不兼容,要重新下载

二、安装PyTorch+TorchVision+Torchaudio

PyTorch是深度学习的「核心框架本体」,而TorchVision、Torchaudio是 PyTorch 官方出品、无缝兼容的「领域专属工具库」------ 前者专攻计算机视觉(CV),后者专攻音频 / 语音处理,三者合称 PyTorch「视觉 + 音频」全家桶,搭配使用能省去大量底层代码编写,是 CV 和语音深度学习的标配组合

这三个工具下载过程都一样,以Torch为例:

首先进入Torch官网:

https://pytorch.org/

进入这个界面,下滑

选择系统,编译语言等等,将红框中的链接复制打开(此链接为torch以及其中所有库的所有版本)

进入此界面,找到Torch,点击进去

使用快捷键ctrl+f键输入想要下载的版本,要对应自己下载的CUDA版本与python版本,和自己的电脑系统

可能会有些慢,要耐心等待

下载完成在文件夹中打开,选中右击,选择复制文件地址

在命令提示符中输入

pip install 复制的文件地址

在最后出现successful就成功了

其中黄色的字体是警告,不是报错~

TorchVision、Torchaudio下载过程与Torch过程一样

如果在下载TorchVision、Torchaudio中torch出现问题,将安装torch语句在输一遍就行了

到这里,深度学习核心环境------CUDA+PyTorch+TorchVision+Torchaudio就全部安装完成了。整个过程的核心关键点只有两个:一是CUDA版本要匹配自己的显卡驱动,二是PyTorch、TorchVision、Torchaudio的版本要相互对应,且适配已安装的CUDA版本,这也是避开绝大多数安装坑的关键。安装完成后,建议大家按照前文提到的验证方法,确认CUDA和PyTorch均能正常运行,尤其是GPU加速功能(torch.cuda.is_available()返回True),这样后续训练模型才能实现效率翻倍。如果安装过程中遇到卡顿、报错,可回头核对每一步的操作细节,重点检查版本匹配和环境变量配置。环境搭建完成,就可以放心开启你的深度学习实操之旅啦!

相关推荐
行走的小派1 分钟前
本地跑模型+原生开源鸿蒙:拆解香橙派AI手机的12TOPS端侧硬核玩法
人工智能·开源·harmonyos
2501_948114241 分钟前
从 Claude Code 源码泄露看 2026 年 Agent 架构演进与工程化实践
大数据·人工智能·架构
小悟空1 分钟前
[AI生成]Iceberg 更新操作技术调研报告
人工智能
hughnz3 分钟前
断钻具的原因与预防
人工智能·钻井
Legend NO246 分钟前
数据资产评估风险识别、分析与管控体系建设
大数据·人工智能·python
Bill Adams14 分钟前
如何基于Harness Engineering设计一个Agent OS
人工智能·prompt·agent·智能体·harness
毕胜客源码17 分钟前
改进yolov8的香蕉成熟度检测系统,改进前后的模型指标对比,有技术文档,支持图像、视频和摄像实时检测
人工智能·python·深度学习·yolo·django
TheRouter17 分钟前
AI Agent 开发中的模型调度策略:何时用便宜模型,何时用强模型
前端·人工智能·react.js
骑着蜗牛找妞19 分钟前
cmux 使用教程:专为 AI Coding 打造的终端工作流引擎
人工智能