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

window下pytorch入门深度学习环境安装与配置

windows下安装pytorch GPU版本

文章目录

    • ***1,确定自己的硬件信息,确定自己的电脑是否有英伟达## 标题(NVIDIA)显卡:**
    • ***2,下载安装Anaconda并进行一些虚拟环境操作:***
    • ***3.接下来是要为安装pytorch做一些关于GPU的准备:***
    • ***3.安装pytorch***
    • ***4.验证pytorch安装成功***
    • **总结:**

主要操作:
1,确定自己的硬件信息,确定自己的电脑是否有英伟达(NVIDIA)显卡。
2,下载Anaconda。
3,利用conda或者pip安装pytorch(我利用的是pip)
安装pytorch的准备:1,确定自己安装的cuda版本与GPU的关系 2,安装GPU版本pytorch

4,验证pytorch是否安装成功
安装流程

*1,确定自己的硬件信息,确定自己的电脑是否有英伟达## 标题(NVIDIA)显卡:

    确定这个很简单,你需要打开【任务管理器】中的性能选项卡点击GPU字样的部分就可以查看,图片实例如下:
    如图中可以看到NVIDIA字样就说明有英伟达显卡

    2,下载安装Anaconda并进行一些虚拟环境操作:

    官网:https://www.anaconda.com/,进入官网后选择如图所示选择Anaconda Distribution,然后选择跳过登记,选择第二张图片的字样直接安装


    结果如下:
    之后是验证Anaconda是否安装成功,首先安装完之后打开Environments中可以看到有一个基础虚拟环境base,如下图:
    接下来是利用Anaconda进行一些虚拟环境的操作,下面是一些命令行
    创建新的虚拟环境:conda create -n 名称 python=你想创建的版本(建议3.8以上方便后面下载pytorch)
    进入先前创建的虚拟环境:conda activate 名称
    查看虚拟环境中的包:conda list
    查看创建的虚拟环境:conda env list
    删除虚拟环境的所有东西:conda remove -n 名称 --all
    确定cuda driver的版本:nvidia-smi
    实例:调用含有Anaconda的命令窗口,输入conda env list,可以看到只有一个base


    在这个基础上创建一个新的虚拟环境名字是xlpytorch,python的版本是3.6(建议大家是3.8),直接输入命令行:conda create -n xlpytorch python=3.6(注意空格),结果如图所示:

    我还创建了一个新的虚拟环境名字为xxik-xl,python的版本为3.10,命令行为conda create -n xxik-xl python=3.10,结果如下图所示:

    3.接下来是要为安装pytorch做一些关于GPU的准备:

    确定需要安装的cuda与GPU的关系:确定自己电脑的算力(这里我们直接安装显卡驱动的最新版本),确保自己的CUDA Driver 版本要>=CUDA Runtime版本,以及CUda runtime version <= cuda driver version,从而确保自己的电脑可以有足够的算力跑代码,算力方面可以去cuda官方的网址去找找,接下来我们来安装显卡驱动最新版本。
    网址:https://www.nvidia.cn/Download/index.aspx?lang=cn,进入之后根据自己的GPU来选择最新版本,我的电脑如下图所示:

    因为我的是笔记本所以选择的是带有Notebooks的

    选择之后直接点击查找,开始下载,我的如下图所示:
    点击下载:

    安装过程中一定要关闭游戏,别问我是怎么知道的,我的壁纸引擎还没搞好/(ㄒoㄒ)/~~,安装过程中的选择默认继续就行,安装地址可以自己确定。
    下图是没有安装前的cuda Driver version(输入命令行nvidia-smi可以查看)可以看到有13.0的字样

    接下来是安装最新版本之后的cuda Driver version可以看到有13.3的字样:

    3.安装pytorch

    根据cuda driver的版本我们就可以去pytorch 的官网去确定cuda Runtime版本
    网址:https://pytorch.org/,进入网址后往下滑有安装的字样,进而选择适合自己的最新版本,因为我的cuda driver为13.3的版本所以我选择的是cuda 13.2的版本(pip安装)(因为cuda driver version要>=cuda runtime version)
    选择完之后我们要在虚拟环境在安装pytorch,先进入先前创建好的虚拟环境,然后将安装命令行复制下来,按回车键就可以安装了,我的命令行是:pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132
    这个命令行就在选择完版本的下面直接复制就好,下面第一个图片是进入先前的虚拟环境,第二个图片是安装完成的截图:


    然后我们可以进入这个虚拟环境看一下:
    可以看到torch 和torch vision的字样,pypi是指通过pip安装的。

    4.验证pytorch安装成功

    相应的一些步骤
    1.激活对应的虚拟环境(你安装的pytorch的虚拟环境)命令行:conda activate 虚拟环境名
    2.输入conda list ,看有没有torch
    3.输入python
    4.输入import torch
    5.输入torch.cuda.is_available()
    6.如果显示True,就说明这个pytorch安装成功了

    总结:

    主要是确定cuda runtime的版本,以及进行cuda driver的最新版本安装,这是我第一次安装成功,将过程分享给大家,希望能帮到大家

    http://www.cnnetsun.cn/news/3625018.html

    相关文章:

  • 3步搞定Video DownloadHelper配套应用安装:告别浏览器扩展无法识别的烦恼
  • AI决策辅助系统:多模态大模型在生活场景中的应用
  • 安全组网前十供应商有哪些
  • OpenClaw智能养殖系统:从机械臂到决策脑的进化
  • WPS演示文稿计算机二级考试全攻略:从基础操作到高级技巧
  • Unity中Protobuf的GC优化实战:对象池与内存管理策略
  • 行业深度|AI Agent全面落地,中小企业短视频营销迎来范式革新
  • TI ADS8353/ADS7853评估板深度解析:从硬件配置到性能测试全流程
  • 5分钟解锁WeMod Pro会员:免费激活完整高级功能终极指南
  • HoRain云--JavaScript 调试
  • 基于YOLOv8与SlowFast的轻量化智能安防实践
  • 肝细胞特异性启动子有哪些?TBG和Alb在AAV肝脏递送中的应用
  • 大语言模型自我博弈:提升推理能力的革命性方法
  • 5分钟视频转PDF:智能提取PPT内容的高效解决方案
  • Linux操作系统核心解析:从命令行到架构设计
  • 如何安全解锁WeMod专业版功能:本地化方案终极指南
  • 基于Android的健康饮食推荐系统任务书
  • 技术写作与知识沉淀方法论:从信息碎片到个人品牌的系统化构建
  • win11安装TortoiseSVN配置svn服务端与汉化
  • 【单片机毕业设计推荐】基于 STM32 的带温度补偿超声波测距预警系统设计,基于 STM32 的可调阈值超声波声光报警装置设计(014203)
  • Shiro漏洞检测工具环境配置:JavaFX与JDK11实战指南
  • Mapbox GL JS 3.27.0 发布:升级 TypeScript 7,修复多项关键 Bug
  • 新闻稿AI化已成定局:但你的团队还在用ChatGPT抄提示词?——企业级新闻生成平台选型评估矩阵(含Latency/FAIR/Traceability三维评分标准)
  • Agentic AI时代:提示工程架构师的技术转型与实战
  • Postman Fuzz 模块 vs APIFox Fuzz 模糊测试 完整对比分析
  • VMware Unlocker 3.0 实战指南:在普通PC上解锁macOS虚拟机支持
  • 多模型路由与推理成本优化:一个网关调度 3 个 LLM 的生产级方案复盘
  • Linux第29篇:Kubernetes从零部署Java微服务集群:容器编排实战
  • OneMore插件:160+功能重塑你的OneNote笔记工作流
  • AI如何动态调节《原神》角色强度?:基于百万局对战数据的实时平衡模型揭秘