news 2026/9/24 17:14:19

深度学习--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/9/23 7:40:55

西门子1200立库机器人码垛机伺服视觉AGV程序大揭秘

西门子1200立库机器人码垛机伺服视觉AGV程序 包括2台西门子PLC1215程序和2台西门子触摸屏TP700程序 PLC与工业相机视觉定位及机器人使用Modbus TCP通讯 PLC和码垛机Modbus TCP通讯(SCL语言) PLC和4台G120变频使用Profinet通讯 1个伺服轴,AGV …

作者头像 李华
网站建设 2026/9/19 3:23:14

基于 MATLAB 的一维数据二分类

基于MATLAB的一维数据二分类在数据分析和机器学习的世界里,二分类问题是最基础也是最常见的任务之一。今天咱们就来聊聊如何使用 MATLAB 对一维数据进行二分类。 问题背景 假设我们有一组一维的数据,这些数据可以是各种测量值,比如温度、压力…

作者头像 李华
网站建设 2026/9/22 8:30:14

基于主从博弈理论的共享储能与综合能源微网优化运行研究

MATLAB代码:基于主从博弈理论的共享储能与综合能源微网优化运行研究 关键词:主从博弈 共享储能 综合能源微网 优化调度 参考文档:《基于主从博弈理论的共享储能与综合能源微网优化运行研究》完全复现 仿真平台:MATLAB yalmipcple…

作者头像 李华
网站建设 2026/9/20 18:15:47

YOLO模型训练成本太高?试试我们的低成本高性能算力方案

YOLO模型训练成本太高?试试我们的低成本高性能算力方案 在智能制造工厂的质检线上,一台搭载AI视觉系统的机械臂正高速运转——它需要在毫秒级时间内识别出电路板上的微小焊点缺陷。这类对实时性与精度双高要求的任务,如今大多由YOLO系列模型驱…

作者头像 李华
网站建设 2026/9/18 16:38:00

YOLO实时性背后的秘密:浅析网格预测与锚框机制

YOLO实时性背后的秘密:浅析网格预测与锚框机制 在智能制造车间的一条高速SMT贴片线上,每分钟有数百块PCB板流过检测工位。摄像头捕捉图像后,系统必须在15毫秒内完成缺陷识别——是虚焊、错件还是缺件?任何延迟都会导致整条产线停摆…

作者头像 李华
网站建设 2026/9/9 9:51:33

异步电机软启动/软起动(调压调速) (基于导通角或者关断角控制的斜坡电压软启动,功率因数闭环软...

异步电机软启动/软起动(调压调速) (基于导通角或者关断角控制的斜坡电压软启动,功率因数闭环软启动)。 提供说明及资料。 异步电机软启动这事儿,说白了就是让电机别一上来就猛冲。直接全压启动的电流冲击能…

作者头像 李华