Windows下安装GPU版Pytorch

升级Driver到最新版本

  1. Windows搜索栏中输入设备管理器
  2. 找到显示适配器一项,点击展开,你将看到你的NVIDIA显卡列在其中
  3. 右键点击你的NVIDIA显卡,选择更新驱动软件…。在弹出的对话框中,选择自动搜索更新的驱动软件。之后,系统将自动搜索、下载并安装最新的驱动版本。完成后,可能需要重启计算机。

查看适配Driver的Cuda版本

  1. 桌面 --》 鼠标右键 --》NVIDIA控制面板,或在任务栏中右击NVIDIA设置打开NVIDIA控制面板
  2. NVIDIA控制面板 –》系统信息

显示中项目可以看到电脑的显卡信息,如 GeForce RTX 2070

  1. 组件 --》NVCUDA64.DLL 在产品名称栏可以看到CUDA driver

在这里插入图片描述

组件中可以看到CUDA 11.6.127 driver,则我的电脑适配的是Cuda 11.6及以下的版本

卸载已安装的Cuda

  1. 设置 --》应用 --》应用和功能
  2. 在搜素中查找 NVIDIA,注意在显示列表中仅卸载含CUDA的软件即可,仅卸载含CUDA的软件即可,仅卸载含CUDA的软件即可,请慎重卸载

没有CUDA相关软件,则不进行卸载步骤

选择对应CUDA

  1. 查看自身显卡的算力,在此网站中搜索显卡型号,可以看到我显卡对应的算力是7.5

在这里插入图片描述

  1. 查看显卡算力支持的CUDA版本

在这里插入图片描述

算力7.5支持的CUDA版本有10.0-10.2,11.0,11.1-11.4,11.5–11.7.1,11.8,12.0-12.4
显卡驱动最高支持的CUDA为11.6,安装的CUDA应该小于等于11.6,因此我可选的CUDA版本有10.0-10.2,11.0,11.1-11.4,11.5-11.6

下载对应Cuda

进入Cuda工具包下载页面,选择适合自己版本的Cuda,我在这里选择11.6.0的版本

在这里插入图片描述

选择系统类型为Windows
选择软件包类别,点击download

在这里插入图片描述

如果官方网页迟迟打不开,可以试着用手机下载,亲测有效

安装Cuda

Cuda默认会安装到C盘/Program Files/NVIDIA Corporation中
因C盘容量不足,准备将Cuda安装在D盘,新建了以下目录文件

在这里插入图片描述

目录结构如下图所示,六个文件夹均为手动新建,此后安装时将路径选在这边

在这里插入图片描述

双击打开之前下载的Cuda 11.6.0的安装包,弹出的界面表示将生成一个临时文件,路径可以修改在其他位置,点击OK

等待进度达到100%,弹出界面后点击同意并继续

由于有些东西是不需要的,因此选择自定义安装,然后选择下一步

勾选如下文两张图所示的内容,或者默认所有组件安装

在这里插入图片描述
在这里插入图片描述

点击下一步,选择安装位置,下面是默认的安装位置,我要安装在D盘因此点击浏览,选择到我们之前新建的文件位置,注意文件名称与官方中的保持一致,修改后如下图所示,然后点击下一步

在这里插入图片描述

打钩,点击Next,完成后如下图所示,点击关闭即可

在这里插入图片描述

还记得之前生成的临时文件吗,这个时候已经自动删除

下载Cudnn

cuDNN已经被集成在Pytorch中,无需单独安装

cuDNN(CUDA Deep Neural Network library)是 NVIDIA 专门为深度学习任务而设计的加速库。cuDNN 提供了一系列高度优化的深度学习基础操作的实现,例如卷积、池化、归一化等,以便在 GPU 上高效执行神经网络的前向和反向传播。cuDNN 通过优化深度学习的基本运算,提高了深度学习框架在 GPU 上的性能。

安装Cudnn

验证Cuda是否安装成功

使用如下命令,出现版本信息即代表安装成功

nvcc -V

在这里插入图片描述

安装Pytorch GPU版

进入Pytorch官网,点击Get Started后下滑进入START LOCALLY界面

如果在START LOCALLY界面对想要下载的 PyTorch 版本进行配置时发现没有合适的 CUDA 版本或者新项目的代码只支持低版本的 PyTorch,这时就需要安装历史版本的 PyTorch

比如,之前步骤确定的需要下载的 CUDA 版本为“11.6”,然而在“START LOCALLY”界面对想要下载的 PyTorch 版本进行配置时发现只有“11.8”和“12.1”的版本,这时就需要下载历史版本

进入Pytorch官网,点击Get Started,然后在出现的界面中选择Previous PyTorch Versions

进入Previous PyTorch Versions界面后下滑,选择包管理器下的Linux and Windows系统,然后选择合适的 CUDA 版本,复制配置好的 PyTorch 版本后面的代码如:

# CUDA 11.6
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

以上代码是用pip安装Pytorch,如果电脑安装了Anaconda,也可以复制conda的代码

在这里插入图片描述

进入Python虚拟环境,如python311env

在这里插入图片描述
在这里插入图片描述

带有CUDA字样的就是GPU版,由于我们已安装CUDA的版本是11.6,因此我们可以下载小于等于11.6的版本
在anaconda中先激活需要安装pytorch的环境,我这里是之前创建过的名为d2l的环境

在这里我要安装1.11.0的Pytorch,选择CUDA11.3的版本

复制该条指令到anaconda中,回车执行

确认安装,输入y

等待下载完成

出现以下界面代表安装成功

使用conda list 查看已安装的包

发现pytorch torchvision torchaudio cudatoolkit均在其中,至此,安装成功

参考资源

[1] nvidia显卡驱动怎么升级?这些方法轻松搞定
[2] 右键没有nvidia控制面板怎么办?
[3] 配置深度学习环境:在Windows(Win10)中安装CUDA,CUDNN,Pytorch GPU版
[4] (2022.4)Win10最新Anaconda安装Pytorch GPU环境(CUDA)教程(极其适合初学者)
[5] 环境配置:选择正确版本的CUDA和PyTorch安装(windows)

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/619997.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【基于HTML5的网页设计及应用】——事件代理.

🎃个人专栏: 🐬 算法设计与分析:算法设计与分析_IT闫的博客-CSDN博客 🐳Java基础:Java基础_IT闫的博客-CSDN博客 🐋c语言:c语言_IT闫的博客-CSDN博客 🐟MySQL&#xff1a…

一文搞定关于SkyWalking告警的那些事儿

Apache SkyWalking告警是由一组规则驱动,这些规则定义在config/alarm-settings.yml文件中。 告警规则 告警规则定义了触发告警所考虑的条件。告警规则有两种类型,单独规则和复合规则,复合规则是单独规则的组合。这里重点介绍一下单独规则&a…

小程序视频怎么保存到本地

掌握视频下载高手的妙招,轻松将微信小程序中的视频内容保存到本地📥。遵循本文步骤,无需繁琐操作,快速实现视频下载,享受随时观看的便捷。 下载高手我已经打包好了 下载高手链接:https://pan.baidu.com/s…

【位运算】Leetcode 丢失的数字

题目解析 268. 丢失的数字 本题的意思就是数组的长度为n,在[0,n]区间中寻找缺失的一个数字 算法讲解 直观思路:排序 Hash,顺序查找缺失的数字 优化:使用异或,首先将[0,n]之间所有数字异或在一起,然后将…

MGRE-OSPF接口网络类型实验

OSPF接口网络类型实验 一,实验拓扑 初始拓扑: 最终拓扑: 二,实验要求及分析 要求: 1,R6为ISP只能配置IP地址,R1-R5的环回为私有网段 2,R1/R4/R5为全连的MGRE结构,R…

LeetCode 349. 两个数组的交集

LeetCode 349. 两个数组的交集 1、题目 力扣题目链接:349. 两个数组的交集 给定两个数组 nums1 和 nums2 ,返回 它们的 交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。 示例 1: 输入:nums1 […

代码随想录算法练习Day13:有效的字母异位词

题目: 给定两个字符串 s 和 t ,编写一个函数来判断 t 是否是 s 的字母异位词。 注意:若 s 和 t 中每个字符出现的次数都相同,则称 s 和 t 互为字母异位词。 题目链接:242.有效的字母异位词 卡哥的视频讲解&#xff…

2024国内外常用药物研发数据库(收藏备用)

几十年前,医药研发领域数据查询可谓是一项繁琐而复杂的工作,研发人员需要耗费大量的时间和精力,穿梭于各类纸质文献、专业期刊和实验报告中,寻找各类宝贵数据。然而随着科技的发展,众多医药专业数据库如雨后春笋般涌现…

mysql 查询实战2-解答

看了mysql 查询实战2-题目-CSDN博客的题目,继续进行解答。 6、查询⽹站访问⾼峰期 目标: 查询网站访问高峰时期,高峰时期定义:至少连续三天访问量>1000 1,关联查询 要连续三天,至少要声明“自身”三个去…

淄博烧烤、天水麻辣烫-《实现领域驱动设计》中译本评点-第2章(5)

相关链接 DDD领域驱动设计批评文集>> 汪峰哭晕在厕所-《实现领域驱动设计》中译本评点-第2章(1) 可不是乱打的-《实现领域驱动设计》中译本评点-第2章(2) “领域”的错误定义-《实现领域驱动设计》中译本评点-第2章&…

关于机器学习/深度学习的一些事-答知乎问(六)

如何使用频率域变换对序列数据进行增强? 时频变换是常见的信号分析思路,同样可用于数据增强。在频率域添加噪声是方法之一。比如可以对传感器信号应用短时傅里叶变换STFT得到具有时序关系的谱特征,再在谱特征上应用两种数据增强方法。一是对…

版本控制工具Git的使用

1、Git的基本概念和使用 1、Git是什么? ● Git: 是一个开源的分布式版本控制系统,可以有效、高速的处理从很小到非常大的项目版本管理。 ● GitHub: 全球最大的面向开源及私有软件项目的托管平台,免费注册并且可以免费托管开源代码。 ● GitLab:与GitHub类似&a…