如何在Windows上配置PyTorch GPU环境 - 详细步骤指南

这篇具有很好参考价值的文章主要介绍了如何在Windows上配置PyTorch GPU环境 - 详细步骤指南。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

如何在Windows系统上配置PyTorch GPU环境的详细步骤。

以下是您所描述步骤的简要概述:

  1. 打开NVIDIA控制面板:通过搜索NVIDIA打开控制面板,并查看系统信息以确认驱动版本。

  2. 下载CUDA:根据NVIDIA驱动版本,从NVIDIA官网下载支持的CUDA版本。

  3. 下载cuDNN:下载与CUDA版本相匹配的cuDNN库,这是用于深度学习的加速库。

  4. 安装CUDA:运行下载的CUDA安装程序,并根据当前系统已安装的CUDA版本选择是否覆盖安装。

  5. 安装cuDNN:将下载的cuDNN文件解压并重命名,然后复制到CUDA目录下,完成cuDNN的安装。

  6. 配置环境变量:在系统属性中配置Path环境变量,添加CUDA和cuDNN的bin目录路径。

  7. 安装Anaconda:如果尚未安装,则需要先安装Anaconda。

  8. 创建新的conda环境:在Anaconda Prompt中创建新的环境,并指定Python版本。

  9. 激活conda环境:激活刚刚创建的conda环境。

  10. 安装PyTorch:访问PyTorch官网,选择与系统配置相匹配的安装命令,并在激活的conda环境中运行该命令。

  11. 验证安装:通过运行特定的Python命令来验证PyTorch和GPU是否安装配置成功。

具体步骤和示例

网上有许多pytorch -gpu环境配置的方法,我结合了许多篇文章自己成功安装并把完整的安装过程记录下来,大家可以参考一下Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
首先打开NVIDIA控制面板:搜索NViDIA即可出现
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
进入NVIDIA控制面板后点击系统信息:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

我的驱动版本支持CUDA 11.6
NVIDIA官网下载CUDA(是NVIDIA推出的用于自家GPU的并行计算框架,CUDA只能在NVIDIA的GPU上运行,):
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

选择对应版本

Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

接下来下载cuDNN(是NVIDIA打造的针对深度神经网络的加速库,不是必须,但一般会采用)

Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

选择对应版本:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

两个东西下好后:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

运行cuda的那个exe文件安装cuda:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

如果当前版本比新版本要大,那么就取消打勾
如果当前版本比新版本要小或者等于,那么勾选上
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

之后进入Dos命令行:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

若有版本信息则安装成功

接下来是cudnn的安装,严格来说不是安装,就是复制粘贴
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

重命名为cudnn
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

接下来复制到
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

cudnn就安装成功了

接下来配置环境变量:
此电脑->右键属性->高级系统设置->高级->环境变量
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

Path路径双击打开
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

可能已经自动配置好了一个环境变量,即上图中的CUDA\v11.6\bin,若没有也可以自己配置。
还有一个环境变量需要配置,如下:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

这个路径也要加到环境变量中:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

复制上面的路径,填入:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

配置好这两个环境变量后,下面是安装pytorch:
首先在conda中新建环境:
首先可以查看自己的python版本:
Win+R,打开cmd,输入python:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

再进入Anaconda Prompt(这个需要先下载Anaconda)
输入:conda create -n pytorchgpu python=3.8
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

之后弹出提示,输入y,即可安装,
安装成功后,输入:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

查看环境,之后激活pytorchgpu环境:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

接下来安装pytorch:
进入pytorch官网(pytorch.org)选择对应安装命令:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

复制下来在激活的pytorchgpu环境中运行:
这是运行过程
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

结束:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建

验证:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建
到此就结束了
如果不放心可以再使用下面命令验证一下:
Pythob pytorch,PyTorch GPU安装,pytorch gpu环境搭建文章来源地址https://www.toymoban.com/news/detail-603884.html

到了这里,关于如何在Windows上配置PyTorch GPU环境 - 详细步骤指南的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Windows环境下GPU版本pytorch安装

    通过官网引导就可以完成安装,或者通过下面网址复制conda安装命令,安装历史版本 Previous PyTorch Versions | PyTorch ​ 输出: 参考: Windows环境下Gpu版本的Pytorch安装_pytorch向下兼容吗_晓码bigdata的博客-CSDN博客

    2024年02月12日
    浏览(53)
  • 【深度学习】【三维重建】windows10环境配置PyTorch3d详细教程

    本人windows10下使用【Code for Neural Reflectance Surfaces (NeRS)】算法时需要搭建PyTorch3d环境,故此以详细教程以该算法依赖的环境版本为参照。 应很多朋友的要求,博主将自己的pytorch3d的conda环境打包放在【百度云盘;提取码:3tyw 】供大家参考,注意环境版本要求保持一致,并不适合

    2024年02月11日
    浏览(53)
  • 如何配置java环境以及tomcat详细步骤

    jdk 官网地址: https://www.oracle.com/java/(opens new window) #1、jdk 下载 进入官网,定位到:Java - Java SE - Oracle JDK 点击进入,如下图所示: 选择 Java archive,再鼠标下拉页面,选择 Java SE 8 (8u202 and earlier) 下载 jdk-8u202-windows-x64.exe

    2023年04月27日
    浏览(49)
  • Windows 系统从零配置 Python 环境,安装CUDA、CUDNN、PyTorch 详细教程

    进入anaconda官网:https://www.anaconda.com/ 点击 download 下载文件,我这里是 Anaconda3-2022.10-Windows-x86_64.exe (后续更新版本exe文件会有差别) 下载后打开 .exe 文件下载 anaconda: 选择安装路径(用默认的路径也可以): 这里两个都选: 然后安装就可以了。 打开 cmd,输入 conda(如果是

    2024年02月03日
    浏览(97)
  • 从零开始系列(1)—— Anaconda + PyCharm + PyTorch(GPU) + 虚拟环境(全步骤,多图警告)

    感谢 姜小敏 同学对我的支持、鼓励和鞭策! 进入 Anaconda 下载界面 ,单击 Download 下载。 下载好之后,进行 安装 。 安装路径 全英文 。 不勾选 第二项。 接下来配置 环境变量 。 基于自定义的 安装路径 ,添加以下三个路径。 将以下图标从 开始菜单 中拖拽到 桌面 上,方便

    2023年04月09日
    浏览(41)
  • 如何在Windows中配置多个显示器?这里提供详细步骤

    Windows可以通过多种方式使用多个显示器,扩展或复制主显示器。你甚至可以关闭主显示器。以下是如何使用简单的键盘快捷键更改辅助显示设置。 要快速更改Windows 10处理多个显示器的方式,请按Windows+P。屏幕右侧会弹出一个名为“投影”的深灰色菜单。 从这里,你可以使用

    2024年02月20日
    浏览(56)
  • windows11+GPU1060安装强化学习环境之pytorch

    这里大家看一个视频系列,讲得非常详细,链接在此:https://www.bilibili.com/video/BV1S5411X7FY?p=28vd_source=3be739b673e1151850f4b8060ac78e1a 这里主要是说我遇到的问题以及解决办法。 首先,我的笔记本是有显卡的,只不过算力为6,较低,但是已经差不多了,主要是想在windows上操作,嗯,还

    2024年01月19日
    浏览(47)
  • 配置pytorch(gpu)分析环境

    Pytorch是目前最火的深度学习框架之一,另一个是TensorFlow。不过我之前一直用到是CPU版本,几个月前买了一台3070Ti的笔记本(是的,我在40系显卡出来的时候,买了30系,这确实一言难尽),同时我也有一台M1芯片Macbook Pro,目前也支持了pytorch的GPU加速,所以我就想着,在这两

    2023年04月08日
    浏览(41)
  • pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南

    🌷🍁 博主 libin9iOak带您 Go to New World.✨🍁 🦄 个人主页——libin9iOak的博客🎐 🐳 《面试题大全》 文章图文并茂🦕生动形象🦖简单易学!欢迎大家来踩踩~🌺 🌊 《IDEA开发秘籍》学会IDEA常用操作,工作效率翻倍~💐 🪁🍁 希望本文能够给您带来一定的帮助🌸文章粗浅,敬

    2024年02月16日
    浏览(87)
  • Windows安装GPU环境CUDA、深度学习框架Tensorflow和Pytorch

    Windows安装GPU环境CUDA、深度学习框架Tensorflow和Pytorch 首先需要安装GPU环境,包括cuda和cudnn。 深度学习本质上就是训练深度卷积神经网络。 cuda:显卡能够完成并行计算任务,所有的操作是比较底层的、复杂的。 cudnn:在cuda之上有一个专门用于深度神经网络的SDK库来加速完成相

    2023年04月26日
    浏览(55)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包