PyTorch深度学习——安装PyTorch,Cuda、CuDNN 配置环境

Windows虚拟环境下pytorch、cuda不同版本配置问题 不同项目需要使用不同的配置环境,踩坑踩了两天,一直没搞清楚虚拟环境要不要安装cuda,后来才知道cuda是向下兼容的。 torch匹配cuda的本质是匹配cudatoolkit 与 cudnn 而高版本的cuda是可以向下兼任cudatoolkit和cudnn的 所以创建新环境安装项目需要的torch即可,会有对应torch版本所需要的cudatoolkit被安装(cudatoolkit是cuda的子集,里面的东西够pytorch使用),这样使用的cuda就不会是你系统的cuda11.1了。 前提:系统 阅读详情

PyTorch的下载

没有选择下载Anaconda的方法,而是选择分别下载Cuda、CuDNN,而后利用pip下载PyTorch。

PyTorch具体的下载步骤见一下链接:

Pytorch(一) Pytorch 的安装_开开心心-优快云博客_pytorch

其中,由于以上链接Cuda和CuDNN不太清晰,针对Cuda、CuDNN安装教程补充见以下链接:

cuda安装教程+cudnn安装教程_sinat_23619409的博客-优快云博客_cuda安装

针对PyTorch下载补充:

  1. 进入PyTorch官网Start Locally | PyTorch
  2. 按照图中选择 pip下载方式

 选择复制箭头所指命令,在cmd控制面板中输入命令,而后开始下载

        3:下载成功后,在PyCharm中进行测试,输入如下命令

import torch

print("torch version{}".format(torch.__version__))

 运行出现如图所示结果,说明安装成功


以下为PyTorch、Cuda、CuDNN简介

PyTorch:

       PyTorch是一个开源的Python机器学习库,其前身是著名的机器学习库Torch。2017年1月,由Facebook人工智能研究院(FAIR)基于Torch推出了PyTorch,它是一个面向Python语言的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络,这是很多主流深度学习框架比如Tensorflow等都不支持的。PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。除了Facebook外,它已经被Twitter、CMU和Salesforce等机构采用。作为经典机器学习库Torch的端口,PyTorch 为 Python 语言使用者提供了舒适的深度学习开发选择。

Cuda:

        随着显卡的发展,GPU越来越强大,而且GPU为显示图像做了优化。在计算上已经超越了通用的CPU。如此强大的芯片如果只是作为显卡就太浪费了,因此NVidia推出CUDA,让显卡可以用于图像渲染和计算以外的目的(例如这里提到的通用并行计算)。CUDA即Compute Unified Device Architecture,是NVidia利用GPU平台进行通用并行计算的一种架构,它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。开发人员可以利用C言、OpenCL、Fortran、c++等为CUDA架构编写程序。简单来理解,cuda就是NVidia提供的可以将显卡进行并行运算的一种软件驱动。

CuDNN:

       我们使用Pytorch的最终目标是用它来进行深度学习训练和推理,深度学习本质上就是训练深度卷积神经网络。尽管我们已经可以用cuda使得显卡来完成并行计算任务,但是所有的操作还是比较底层的、复杂的。cudnn就是在cuda之上有一个专门用于深度神经网络的SDK库来加速完成相关特定的深度学习操作。

       NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如caffe、tensorflow、pytorch、mxnet等。cudnn简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是调整性能,同时还可以在GPU上实现高性能现代并行计算。

       简单来说,cuda就是用来定义显卡并行运算的一些列底层GPU操作库,cudnn则是在cuda基础上专门正对深度学习定制的高级GPU操作库。

简介部分引用链接 面向Windows的Pytorch完整安装教程_冰海的博客-优快云博客_windows安装pytorch中的内容。

完美解决(最简单):不同虚拟环境不同CUDA版本的安装问题(tensorflow-gpu与pytorch) 一、虚拟环境1:torch环境的cuda配置 1.先安装Torch 进入pytorch官网,https://pytorch.org/get-started/locally/找到想要安装的torch及对应的cuda,(比如我要下载cuda10.2的torch),选中cuda10.2对应的torch(如果没有想要的,查看以前版本torch),复制pip安装的命令,激活虚拟环境输入命令安装即可 2.安装cudnn 安装完后condalist查看安装情况,你会发现通过这个命令会自动安装torch、to. 阅读详情

相关推荐

【RTX 3060Ti 深度学习环境配置图文(安装Anaconda、VScode、CUDA、CUDNN、pytorch)】

环境配置3060ti,cuda 11.4 CUDNN 8.2 pytorch1.9,python3.9

Magize的博客 4643

简单方法适用于pytorch的环境配置cuda及cudnn

然后直接一条命令配置好cuda11.3及cudnn conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge 或者这个命令:cuda11.1及对应的cudnn pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.p

mfjzjhayb的博客 3441

【Ubuntu最新最全深度学习环境配置】Ubuntu24.04 + CUDA12.8 + CUDNN + Python + Miniconda + Pytorch + Pycharm安装及环境变量添加

Ubuntu最新最全深度学习环境配置

taocui_的博客 3290

pytorch-GPU环境配置1:win10如何安装cuda以及切换不同版本的cuda

本文主要介绍怎么安装cuda以及win10中如何切换不同版本的cuda。由于本人经常在多个版本的cuda中切换,因此会用到Anaconda3创建多个环境,因此本文操作就是基于conda来进行的,如果没有Anaconda3可以看一下其他博主的教程。

我的鱼呀的博客 2959

学Simulink--基于多能源系统协同与能源互联网的电网系统场景实例:风光氢储一体化系统的能量管理与调度仿真

在全球“双碳”目标驱动下,高比例可再生能源并网成为电网发展的核心方向。然而,风电与光伏具有强随机性与间歇性,给电网稳定运行带来巨大挑战。风光氢储一体化系统(Wind-Solar-Hydrogen-Storage Integrated System)通过将可再生能源发电电解水制氢氢气储存与燃料电池发电有机结合,构建了一个长时储能与能量形态转换的闭环系统,是实现能源互联网与零碳电力系统的关键路径。✅平抑波动:氢储能可实现跨天、跨周储能✅能量转化:电→氢→电/热/交通燃料,实现多能互补✅长时储能。

amy_mhd的博客 1071

【驭风计划 算法训练营】周测4:习题11-4、8-4、12-4

给出 nn 个非负整数 x1,x2,⋯,xnx1,x2,⋯,xn,有 qq 个询问,第 jj 个询问给出一个非负整数 yjyj,求其中,xor 表示按位异或运算。

Mege1的博客 1043

第二章 pytorch的基础知识

pytorch的基本单元以及自动求导等

m0_49040734的博客 900

深度学习Halcon中GPU显存利用率高,但GPU、CPU利用率较低,导致模型训练速度很慢,通过设置硬件参数的解决方法

深度学习Halcon中GPU显存利用率高,但GPU、CPU利用率较低,导致模型训练速度很慢,通过设置硬件参数的解决方法 前言 在训练自己模型的时候,会遇到GPU显存利用率高,但GPU、CPU利用率较低,导致模型训练速度很慢的问题。这个问题一方面是由于硬件没有设置到最佳工作状态,另一方面是代码中参数没有设置好。最近在跑Halcon DL遇到这个问题,而Halcon不像开源的几个框架那样自由,代码封装的比较多。所以遇到这种问题,只能在参数最优的情况下,通过设置硬件参数来提升性能。 当然,在跑DL之前,无论什么框

ZiLiangQin的博客 9914

2023年11月最新WSL2系统安装及深度学习环境CUDA、CUDNN 和Pytorch配置

WSL2系统创建及深度学习环境CUDA、cuDNN、Pytorch创建及使用

weixin_43971236的博客 8118

泛微Ecology9接口调用全流程:从配置到实战Java代码解析

本文详细解析了泛微Ecology9系统接口的完整调用流程,涵盖安全架构、服务端配置与客户端Java实战代码。重点阐述了基于RSA加密的认证机制、Token管理策略,并提供了企业级集成的核心代码示例与生产环境调优建议,帮助开发者安全、高效地实现系统对接。

m9n0o的博客 837

使用Tansformer分割三维腹部多器官--UNETR实战

不会 transformer 没关系,本教程开箱即用。Tina姐总算对transformer下手了,之前觉得难,因为刚开始学序列模型的时候就没学会。然后就一直排斥学transformer。这两周没什么事,加上MONAI有现成的教程,就打算先跑通后,再学理论。然后,顺利的跑通了代码,再学了一周理论,发现它也不过如此嘛,入门还是很容易的。有同学想了解的transformer的话,可以先看完这个实战教程,如果感兴趣,后续会出一个transformer入门路线。利用纯Transformers作为编码器来学习输入量的

Tina姐的博客 1万+

配置深度学习环境:在Windows(Win10)中安装CUDA,CUDNN,Pytorch GPU版

配置深度学习环境详细教程:在Windows(Win10)中安装CUDA,CUDNN,Pytorch GPU版

几度热忱的博客 8536

Vue2项目实战:vue2-org-tree实现动态组织架构图

本文详细介绍了如何在Vue2项目中使用vue2-org-tree组件实现动态、可交互的组织架构图。内容涵盖从快速安装、基础渲染到核心功能实战,包括水平布局、节点展开折叠、响应式数据绑定、深度自定义节点内容以及事件处理。同时提供了性能优化、样式覆盖等避坑指南,并通过一个完整的企业级Demo展示了如何打造功能丰富的架构图,助力高效开发后台管理系统。

weixin_28040125的博客 522

【赛博炼丹】深度学习软件安装及环境配置:Anaconda、CUDA、cuDNN、PyTorch及PyCharm

Anaconda、CUDA、cuDNN、PyTorch及PyCharm

weixin_54416370的博客 4127

基于Python的股票K线图爬取与形态识别系统

本文详细介绍了一个基于Python的股票K线图爬取与形态识别系统。该系统通过网络爬虫技术获取股票历史数据,使用Matplotlib生成K线图,并应用卷积神经网络(CNN)对经典K线形态进行识别。文章详细阐述了数据采集、K线图生成、CNN模型构建与训练的完整流程,并提供了完整的代码实现。本研究为金融科技领域的毕业设计提供了有价值的参考方案。关键词:Python爬虫、K线图、卷积神经网络、形态识别、股票交易。

2201_76125261的博客 1033

基于树莓派4B与NCNN的YOLOv8高效部署实战:无GPU环境下的C++加速优化全流程指南**

本文详细介绍了在树莓派4B上基于NCNN框架部署YOLOv8目标检测模型的全流程。内容包括:从PyTorch模型导出NCNN格式文件、树莓派环境配置(OpenCV和NCNN编译)、C++实现代码解析(含模型加载、推理及后处理),以及实际部署测试结果。在树莓派4B上达到约2FPS,并提供INT8量化、多线程等优化建议。该方案为无GPU设备提供了轻量化目标检测的实用部署方案,适合边缘计算场景。

FJN110的博客 917

Windows10下安装anaconda、pycharm,CUDA,CUDNN,配置GPU pytorch深度学习环境

Windows10下安装anaconda、pycharm,配置GPU pytorch深度学习环境安装anaconda安装 CUDA、CUDNN安装CUDNN安装GPU版pytorch安装pycharm 安装anaconda anaconda 下载地址:link 安装很简单,按照提示即可,建议不要安装在系统盘©下,但是最好安装在更目录下:比如安装路径为 : D:\Anaconda3 接下来 为了 ...

weixin_42643931的博客 4578

C# WinForm 基于DirectShow.dll编写的程序,可边预览边保存视频,音频和图片

基于微软的DirectShow.dll的二次开发1.自动识别连接的摄像头跟麦克风且可以对其进行选择(默认选中第一个)2.可录制wmv和avi两种格式的视频并且都带有音频,avi可以录制的分辨率和帧数进行选择,wmv默认进行了淸晰度优化如不喜欢可去除代码中有相关提示3.可在预览点开之后进行拍照,也可边录制视频边拍照4.添加了视频和音频的压缩功能(酌情使用)这次二次开发发现C#对DirectShow的相关信息太少,这次做完也不敢藏私共享出来,还望大家指正

RTX3050ti、3060等系列显卡正确安装cuda、cudnn,配置Pytorch深度学习环境(避免由于版本不适应导致重装)

RTX3050ti、3060等系列显卡正确安装cuda、cudnn,完成深度学习在gpu上的训练,提高运算速度

YAMT-WYHBN的BOOK 3万+

efm32 入门.pdf

本PDF共计22页,介绍的是操作驱动API,下载与否自己斟酌。

jdk-8u144-linux-x64.tar.gz.

该资源为搭建Linux系统Java环境用,资源的后缀为tar.gz格式的压缩包

[超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[2]安装Anaconda与CUDA

[超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch),一看就会,看完就懂!

FRANPPER的博客 3701

ET2012新版免狗免安装绿色软件

ET2012新版免狗免安祖昂绿色软件。,新界面,新视觉,

[超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch

超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch(本篇)

FRANPPER的博客 4297

2025最新最全Cuda安装,深度学习环境配置(已踩太多坑)Win10+VS2019+Cuda11.0+Cudnn8.9.7+python3.8+pytorch1.7.0+tensorflow2.10

完美最新的win10本地Cuda深度学习python环境安装。Win10+VS2019+Cuda11.0+Cudnn8.9.7+python3.8+pytorch1.7.0+tensorflow2.10。附全部安装包和详细安装流程,以及安装常见问题,版本对应关系等。

qq_45377297的博客 8941

2023年深度学习环境配置Anaconda+Pycharm+CUDA+cudnn+PyTorch踩坑后安装总结

安装CUDA时要注意不要安装最新版的,因为很可能你安装好最新版的CUDA后发现没有合适的额PyTorch源,最好安装能在你电脑出厂日期之后不久的版本,防止安装太老的版本不支持你的显卡,所以我选择了安装CUDA11.3版本。然后按视频进行选择安装版本时,不要用他视频里的百度云里的代码,需要到官网找到对应的代码。接下来安装PyTorch是坑最多的,网上相关视频也很多,最后觉得比较靠谱的是这个。然后把这三个文件夹复制到cuda的安装位置合并就可以,不会替换文件。然后就可以成功安装了。装了一整天终于装上了。

theodore0303的博客 737

cuda,cudnn,Anaconda安装pytorch,配置深度学习环境

深度学习环境配置

weixin_62067430的博客 581

深度学习环境配置:CUDA、cuDNN 和 PyTorch 版本的选择与搭配指南、ubuntu安装cudnn

深度学习环境配置:CUDA、cuDNN 和 PyTorch 版本的选择与搭配指南(建议收藏)_cuda和cudnn版本匹配-优快云博客https://blog.youkuaiyun.com/weixin_58283091/article/details/133349022

weixin_44012667的博客 455

【一文搞懂—防止踩坑】深度学习环境配置:CUDA、cuDNN、算力 和 PyTorch 版本选择、详细教程案例(仔细阅读,建议收藏)

【一文搞懂—防止踩坑】深度学习环境配置:CUDA、cuDNN、算力 和 PyTorch 版本选择、详细教程案例(仔细阅读,建议收藏)

cqy阳的博客 1万+

一行命令解决 深度学习环境 TensorFlow PyTorch Keras安装 cuda cudnn 配置

估计你已经被深度学习环境的安装和配置整得焦头烂额了,cuda和cudnn的安装配置真的好麻烦,下面我将告诉你只要用一行代码就可以解决这个问题。 不过首先,你得下载安装anaconda环境,而不是使用纯Python软件。 然后创建一个你想要运行深度学习的环境。 一般使用过Python的人,以上两步都已经完成,新手的话可以自行查阅如何解决,很简单。 最后。。。。。。 进入终端,进入自己的环境。。。。。...

四个菜 571
下一篇: PyTorch深度学习——线性模型 Python实例(B站刘二大人P2学习笔记)
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付:10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

学习CV的研一小白

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值