news 2026/4/15 17:26:44

深度学习--CUDA安装配置、pytorch库、torchvision库、torchaudio库安装

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
深度学习--CUDA安装配置、pytorch库、torchvision库、torchaudio库安装

一、下载CUDA

1、什么是CUDA

CUDA 是NVIDIA 为自家 GPU 打造的“计算引擎”,它让 GPU 不仅能处理图形,更能变成一个超级并行处理器,用来加速科学计算、人工智能、模拟等海量计算任务。

2、查看电脑版本号

打开终端输入nvidia-smi查看

3、下载CUDA

1)打开链接

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

找到对应版本点击打开,这里因为电脑版本是12.3,所以我现在的是12.1.0版本

2)选择系统和版本,这里选择的是window版本

3)下载成功后,找到下载位置,点击打开

4、安装CUDA

1)双击文件打开显示此界面

2)选择自定义安装

3)取消选项Visual Integration

如果计算机上已经安装了与CUDA版本相兼容的Visual Studio Integration文件,或者通过其他方式(如单独安装Visual Studio的插件)已经集成了CUDA支持,那么在安装CUDA时勾选“Visual Studio Integration”可能会导致冲突或重复安装,进而引发问题。

4)使用默认安装地址

5)安装成功

6)配置环境变量

右击此电脑,选择属性,下拉找到高级系统设置

点击打开

找到环境变量,点击打开,查看系统变量中是否存在圈中的信息

或者在终端输入密命令set cuda查看

7)查看是否安装成功

在命令提示符内输入命令nvcc --version 或者nvcc -V,查看是否有cuda,如果有则安装成功,如果没有则可能版本不兼容

注:在查看的时候一定要关闭终端重新打开,才能查看到是否安装成功

二、下载pytorch包

1、什么是torch

Torch是一个开源的机器学习库,主要用于构建深度学习模型。它基于Lua编程语言,并提供了一个强大的张量计算库,可以有效地进行数值计算和矩阵操作。Torch同时也提供了丰富的神经网络模块,使得用户可以方便地构建和训练深度神经网络模型。

Torch的核心是它的张量库,它支持多维张量的操作和计算,类似于NumPy库。张量是一种多维的数组,可以用于存储和处理各种数据类型的数据。Torch的张量库提供了丰富的函数和运算符,可以进行各种数学运算、线性代数运算和张量操作等。

2、下载torch

进入pytorch官网,选择好需要的类别,然后将下列链接复制到浏览器打开,选择版本进行下载,因为torch所占内存太大,直接用其所给命令去命令提示符下载如果网络稍微有波动则会导致下载失败,所以直接去源网页下载,下列是官网入口:
https://pytorch.org/

1)复制上述链接
https://download.pytorch.org/whl/cu126

下滑找到torch,点击进入

2)找到相应的版本

扩展:怎么查下载的python版本号

点击win+R打开输入框,输入cmd,打开终端,在终端中输入where python,就可以查看下载的所有python版本以及所在的位置

输入python查看正在使用的python版本

可以看到我使用的python版本是3.9版本

2)找到下载的位置

右击此文件,点击复制文件地址

3)安装库

在终端里输入pip install 文件地址

4)查看是否安装成功

输入pip list查看下载的所有库

下滑找到torch、证明下载成功

3、torchaudio库是什么

torchaudio是一个基于PyTorch的音频处理库。它提供了一系列用于加载、转换和处理音频数据的函数和类。torchaudio支持多种音频格式,包括wav、mp3、flac等,并提供了一些基本的音频处理功能,如音频剪辑、音频格式转换等。通过torchaudio,用户可以方便地在PyTorch中进行音频数据的处理和建模。

4、下载torchaudio库

重复上述安装torch的步骤即可,选好版本号

点击,然后快捷键搜索相应版本,操作和torch基本一样

安装方式也是和torch库的安装方式一样

5、torchvision库是什么

torchvision是一个在PyTorch中提供了一些用于图像处理任务的工具的库。它包含了常用的数据集加载、图像变换以及预训练模型等功能。torchvision可以大大简化图像处理任务的开发流程。

6、下载torchvision

步骤和上述一致,注意版本号,最后得到下列即为安装成功
使用pip install查看三个库是否都安装成功

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/14 22:08:20

YOLO模型失败案例复盘:一次因数据偏差导致的事故

YOLO模型失败案例复盘:一次因数据偏差导致的事故 在某电子制造工厂的一条SMT生产线上,自动化质检系统突然“失明”——连续三天未能识别出一批存在明显电容缺失的PCB板。这些本应被拦截的不良品最终流入后续工序,造成数千元损失和客户投诉。而…

作者头像 李华
网站建设 2026/4/15 15:06:26

YOLO目标检测API设计规范:构建易用服务接口的原则

YOLO目标检测API设计规范:构建易用服务接口的原则 在智能制造、智慧城市和自动驾驶等前沿领域,视觉感知正从“可有可无”走向“核心驱动”。面对海量视频流与实时决策需求,如何将强大的AI模型转化为稳定可靠的服务能力,成为工程落…

作者头像 李华
网站建设 2026/4/15 6:55:35

工程实践:破解智能体错误的长尾效应——论“悔改机制”中的通知分级与防再犯设计

在真实业务里,智能体最危险的失败模式往往不是“当场答错”——因为当场答错至少还有机会被用户质疑、被客服兜底、被人工复核流程拦住。更隐蔽、也更具破坏性的情况是:智能体在某一次会话里给出了看似可信的建议,用户照做了,流程…

作者头像 李华
网站建设 2026/4/15 7:35:17

YOLO模型安全防护指南:防止恶意输入攻击的实践建议

YOLO模型安全防护指南:防止恶意输入攻击的实践建议 在智能制造车间的视觉质检线上,一台搭载YOLOv8的边缘设备突然开始将所有缺陷产品标记为“合格”——调查发现,攻击者通过监控摄像头注入了一组经过精心扰动的图像,成功欺骗了检测…

作者头像 李华
网站建设 2026/4/15 10:24:00

YOLO模型加密保护方案:防止知识产权泄露的措施

YOLO模型加密保护方案:防止知识产权泄露的措施 在智能制造、自动驾驶和智能安防等领域,AI模型正从“技术实验品”快速演变为企业核心竞争力的关键组成部分。尤其是像YOLO这类高性能实时目标检测模型,其训练成本高昂、调优过程复杂&#xff0c…

作者头像 李华
网站建设 2026/4/15 16:35:36

2025年大模型架构演变全解析:从GPT到DeepSeek-V3,万字干货必收藏!

文章详细分析了2025年主流开源大模型架构演变,包括DeepSeek-V3/R1、OLMo 2、Gemma 3、Llama 4等。重点探讨了多头潜在注意力(MLA)、混合专家(MoE)、滑动窗口注意力等创新设计,以及归一化层放置等差异。这些优化在保持模型性能的同时,显著提升…

作者头像 李华