深度学习环境搭建: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),这样后续训练模型才能实现效率翻倍。如果安装过程中遇到卡顿、报错,可回头核对每一步的操作细节,重点检查版本匹配和环境变量配置。环境搭建完成,就可以放心开启你的深度学习实操之旅啦!

相关推荐
袁哥大话安全5 小时前
巡隐WEBSHELL扫描软件
人工智能·安全·web
论文复现现场5 小时前
8卡4090能跑70B吗?Llama-2显存预算、QLoRA与通信瓶颈
人工智能·深度学习·分布式训练·llama·显存·qlora·算家云
酷虎软件5 小时前
如何用AI创作AI歌曲AI音乐
人工智能
小宋10215 小时前
A2UI从零实战:Agent安全生成交互界面与事件回传
javascript·人工智能·安全·交互
罗小罗同学5 小时前
谷歌团队在Nat Med最新发表的医学多模态模型,4B参数的模型性能逼近671B的DeepSeek
人工智能·医学图像处理·医工交叉·医学ai
GlobalInfo5 小时前
2026年推理算力超越训练算力,市场调研该关注什么
大数据·人工智能·ai·芯片
陈然信息站5 小时前
PCB覆膜检测场景下明治ESE-10N色标传感器技术适配性分析
人工智能
阿里云大数据AI技术5 小时前
阿里云 Milvus 自研内核 EAGLE 发布:向量检索的SOTA,我们决定自己造
人工智能
码流子5 小时前
04-AI标注系统
人工智能
心易行者5 小时前
Agent应用+API端点商业化进阶实战:从单体智能体到可付费调用的API全流程
运维·服务器·人工智能·python·apache