当前位置: 首页 > news >正文

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

一、下载CUDA

1.什么是CUDA

CUDA(Compute Unified Device Architecture)是由NVIDIA推出的并行计算平台和编程模型,允许开发者利用NVIDIA GPU的并行计算能力加速计算密集型任务。它通过扩展C/C++等语言,提供直接访问GPU虚拟指令集和并行计算元素的接口,允许开发者利用NVIDIA GPU的并行计算能力加速计算密集型任务。它通过扩展C/C++等语言,提供直接访问GPU虚拟指令集和并行计算元素的接口。

2.查看电脑支持版本号

打开命令提示符,输入命令nvidia-smi,即可查看电脑支持的最高版本的版本号

如下图所示:

3.下载CUDA包

下面网址是CUDA的官网

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

选择合适的版本

得到如下图的画面,有非常多的版本,最好选择低于电脑版本的安装,防止有过高的版本不稳定的情况。

选择相应的版本后,点击,再选择自己电脑的系统版本进行安装

4.安装CUDA

打开下载好的文件,选择同意并继续,再选择自定义安装

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


使用默认安装位置,不要随意更改,以防后期使用出现问题

5.配置环境变量

再此电脑中找到高级系统设置,打开并点击环境变量

查看是否有环境变量,没有的话需要手动增加,增加内容为下图所示内容,点击新建添加即可

6.测试是否安装成功

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

注意:可能会遇到不是此命令等错误,可以查询大模型,主要操作就是在相应位置添加路径,配置好环境变量即可

二、下载pytorch包

1.什么是torch

Torch 是一个开源的机器学习框架,最初由纽约大学和 Facebook 的人工智能研究团队开发。它基于 Lua 编程语言,专注于提供高效的数值计算和深度学习功能。Torch 的核心是一个名为torch的库,支持张量计算、自动微分和 GPU 加速。

Torch 的主要特点

  • 张量计算:提供多维数组(张量)操作,类似于 NumPy,但支持 GPU 加速。
  • 自动微分:通过autograd模块实现梯度计算,简化神经网络训练。
  • 模块化设计:包含预定义的神经网络层、损失函数和优化器。
  • 跨平台支持:可在 CPU 和 GPU 上运行,支持 Linux、macOS 和 Windows。

2.下载torch

下面链接是pytorch官网,去官网进行下载

https://pytorch.org/

选择好对应的版本后,从http开始复制一直到结束,用浏览器打开该网址

打开后下拉找到torch

找到你需要的版本

解释:

前面的1.13.1表示版本号,后面的cu117表示cuda的版本号,cp表示python版本号,后面表示的是电脑系统,可以通过再命令提示符输入where python查看python版本

注意:cuda版本可以比安装的版本低,但是不能比安装的版本高,同时要使用你自己相同的python版本

安装库

将下载好的文件,复制其文件地址,再用命令提示符里进行下载

如:pip install "E:\torch-1.13.0+cu117-cp37-cp37m-win_amd64.whl"(文件地址为自己相应的文件地址)

检查是否安装成功

在命令提示符里输入pip list 即可

3.下载torchaudio库

什么是torchaudio库

torchaudio是PyTorch生态系统中的音频处理库,专门为深度学习任务设计。它提供高效的音频加载、预处理、特征提取等功能,并与PyTorch的张量操作无缝集成,支持GPU加速。

核心功能

  • 音频I/O:支持多种格式(如WAV、MP3、FLAC等)的加载与保存。
  • 信号处理:包括重采样、频谱图生成(如Mel频谱)、时域操作(裁剪、填充)等。
  • 数据增强:添加噪声、时移、变速变调等增强音频数据。
  • 预训练模型:集成Wav2Vec2、HuBERT等模型,支持迁移学习。

下载安装

重复上述安装torch的步骤,选对版本就行

点开torchaudio,然后快捷键搜索相应版本,版本号和torch一致即可,下图仅为举例

安装方式也是同样使用pip install 文件地址即可

4.下载torchvision库

什么是torchvision库

torchvision是PyTorch的官方扩展库,专注于计算机视觉任务。它提供了一系列工具和预训练模型,用于图像和视频数据的加载、预处理、增强及模型训练。torchvision与PyTorch深度集成,简化了计算机视觉项目的开发流程。

核心功能模块

数据集与数据加载
torchvision.datasets 包含常用视觉数据集(如CIFAR-10、ImageNet、MNIST),支持直接下载和加载。搭配torch.utils.data.DataLoader可实现高效批量读取。

图像变换与增强
torchvision.transforms 提供多种图像预处理方法,包括裁剪、翻转、归一化等。支持组合多个变换形成处理流水线

下载安装

步骤与上述一致,注意版本号一致,最后得到下列就是安装成功了

http://www.jsqmd.com/news/135556/

相关文章:

  • WiseAgent智能体框架实战之CrewAI篇(四) - 优化智能体的问答能力与记忆系统
  • 建议收藏!2025最新论文降AI率保姆级攻略,学生党必看。
  • Hadoop - 资源调度器YARN和计算引擎MapReduce/Tez/Spark之间是什么关系?
  • 【计算机毕业设计案例】基于Springboot+Vue党员教育和管理系统基于springboot的高校党员信息管理系统(程序+文档+讲解+定制)
  • 基于深度学习的蘑菇种类识别系统的设计与实现(源代码+文档+PPT+调试+讲解)
  • Anthropic 开源 Bloom:基于 LLM 的自动化行为评估框架
  • 超越RLVR陷阱:从设计“奖励契约”到构建“AI宪法”的架构思想
  • Linux:awk升级到5.0.3最新版本(源码编译升级方式)
  • 基于深度学习的淘宝用户购物可视化与行为预测系统设计(源代码+文档+PPT+调试+讲解)
  • 2025最新!10个AI论文网站测评:本科生写论文救星大公开
  • ModelEngine AI Agent通过Nexent 是一个开源智能体SDK和平台打造全能搜索助手
  • 计算机Java毕设实战-基于springBool+Vue小吃美食分享平台的设计与实现【完整源码+LW+部署说明+演示视频,全bao一条龙等】
  • 长亭推出工程级AI开发平台MonkeyCode,开启AI工程级开发新模式
  • 【计算机毕业设计案例】vue和springboot框架开发的户外救援系统基于springboot的户外救援系统(程序+文档+讲解+定制)
  • 基于深度学习的图书推荐系统(源代码+文档+PPT+调试+讲解)
  • 6-10 WPS JS宏 映射应用
  • 完整教程:学算法总换设备?Hello-Algo+cpolar 让学习进度随身带
  • 敏捷咨询:从落地到深耕的全流程赋能之路
  • XML DOM
  • 基于SpringBoot的社区诊所在线挂号与排队应用系统毕业设计项目源码
  • Redis 集群模式Redis Cluster
  • AngularJS 模块
  • 完整教程:50天精通FPGA设计-总体规划
  • Java毕设项目推荐-基于springboot的实验室实验报告管理系统的设计与实现基于SpringBoot和Vue的实验报告管理系统的设计与实现【附源码+文档,调试定制服务】
  • 【工具】log-lottery最受欢迎3D球体年会抽奖程序
  • 中文编程语言“华文 (HuaWen)”综合设计方案框架
  • 基于python的中文起点网top500小说数据提取的设计与实现(毕设源码+文档)
  • C# 字符串比较与比较器应用指南(AI笔记)
  • 读人机沟通法则:理解数字世界的设计与形成05机器可以被测量
  • [晕事]今天做了件晕事97,强制停止ctrl+c make