Windows下安装GPU版Pytorch

news2024/11/14 6:51:12

升级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.coloradmin.cn/o/1595820.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

Scrapy框架 进阶

Scrapy框架基础Scrapy框架进阶 【五】持久化存储 命令行:json、csv等管道:什么数据类型都可以 【1】命令行简单存储 (1)语法 Json格式 scrapy crawl 自定义爬虫程序文件名 -o 文件名.jsonCSV格式 scrapy crawl 自定义爬虫程…

二叉树的链式存储——补充

/二叉树的结点(链式存储) typedef struct BiTNode{ElemType data; //数据域struct BiTNode*lchild,*rchild //左、右孩子指针 }BiTNode,*BiTree;如下图进行存放: 所以,一个二叉树有n个结点的话,那总共就会有2n个指针域…

怎么做预约小程序_探索我们的全新预约小程序

在繁忙的现代生活中,无论是想预约一次美容护理,还是预定一家心仪的餐厅,亦或是安排一次专业的咨询服务,我们都希望能够在最短的时间内完成这些操作,节省时间和精力。如今,一款全新的预约小程序应运而生&…

鸿蒙语言TypeScript学习第16天:【类】

1、TypeScript 类 TypeScript 是面向对象的 JavaScript。 类描述了所创建的对象共同的属性和方法。 TypeScript 支持面向对象的所有特性,比如 类、接口等。 TypeScript 类定义方式如下: class class_name { // 类作用域 }定义类的关键字为 class&am…

AI天使汇联合150家顶级基金、战投,征集优秀AI创业项目

鉴于AI天使汇主办的2024年3月期优秀项目征集活动效果超出预期,3月活动最后TOP20路演者中已有多家快速拿到了TS。 路演活动质量受到了AI创业公司和基金/战投伙伴的高度评价,现在开始四月期活动报名! 本期征集活动联合的顶级基金和战投数量增加到了150家…

MGRE-OSPF接口网络类型实验

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

儿童护眼台灯怎么选?五款必选的高口碑护眼台灯推荐

儿童台灯,想必大家都不会陌生了,是一种学生频繁使用的小灯具,一般指放在桌面用的有底座的电灯。随着近年来儿童青少年的视力急速下滑,很多家长都会选择给孩子选择一款合适的护眼台灯,以便孩子夜晚学习能有个好的照明环…

Stable Diffusion教程:LoRA模型

LoRA模型是一种微调模型,它不能独立生成图片,常常用作大模型的补充,用来生成某种特定主体或者风格的图片。 下载模型 在模型下载网站,如果模型是LoRA模型,网站会特别标识出来。以 liblib.ai为例: 模型左…

three.js捋文档的记录笔记(六):场景 几何体 材质 物体 相机 渲染器的简单理解

三维场景Scene const scene new THREE.Scene();物体形状:几何体 Geometry //创建一个长方体几何对象Geometry const geometry new THREE.BoxGeometry(100, 100, 100); 物体外观:材质Material //创建一个材质对象Material const material new THREE.M…

Spring Boot 学习(3)——Spring Initializr 创建项目问题解决

产生问题的原因,各种的版本都较老,所以导致出现问题。目前暂未打到合适的教程,按老教程学起来先。 小白瞎学,大神勿喷! 再次强调环境:maven 3.3.9、jdk 1.8、idea 2017、Spring 4.3.13、Spring Boot 1.5.…

最优算法100例之47-从尾到头打印单链表

专栏主页:计算机专业基础知识总结(适用于期末复习考研刷题求职面试)系列文章https://blog.csdn.net/seeker1994/category_12585732.html 题目描述 从尾到头打印单链表 题解报告 方法1:头插法逆置单链表然后依次打印;注意此处是不带头结点的单链表,带头节点的操作稍微有…

ELK企业级日志分析系统(elasticsearch+logstash+kibana)

目录 一.ELK概述 1.定义 (1)ElasticSearch (2)Kiabana (3)Logstash (4)Filebeat 2.filebeat结合logstash带来好处 3.为什么要是用ELK? 4.完整日志系统基本特征 …

c++ - 类的默认成员函数

文章目录 前言一、构造函数二、析构函数三、拷贝构造函数四、重载赋值操作符五、取地址及const取地址操作符重载 前言 默认成员函数是编译器自动生成的,也可以自己重写,自己重写之后编译器就不再生成,下面是深入了解这些成员函数。 一、构造…

#381. 四边形继承练习

太爽了 甚至还现学了叉积判断线段是否相交和求面积的方法 先给出我的代码&#xff1a; #include <iostream> #include <vector> #include <iomanip> #include <cmath>using namespace std;//下面需要补充多个类的声明及实现代码 const double EPS 1…

STM32之FreeRTOS移植

1.FreeRTOS的移植过程是将系统需要的文件和代码进行移植和裁剪&#xff0c;其移植的主要过程为&#xff1a; &#xff08;1&#xff09;官网上下载FreeRTOS源码&#xff1a;https://www.freertos.org/ &#xff08;2&#xff09;移植文件夹&#xff0c;在portable文件夹中只需…

2024年文化、历史与人文艺术与社会发展国际会议(CHHASD2024)

2024年文化、历史与人文艺术与社会发展国际会议(CHHASD2024) 会议简介 2024年国际文化、历史、人文、艺术与社会发展会议&#xff08;CHHASD2024&#xff09;将在中国武汉举行&#xff0c;主题为“文化、历史&#xff0c;人文、艺术和社会发展”。CHHASD2024汇集了来自世界各…

【C语言】指针篇-初识指针(1/5)

&#x1f308;个人主页&#xff1a;是店小二呀 &#x1f308;C语言笔记专栏&#xff1a;C语言笔记 &#x1f308;C笔记专栏&#xff1a; C笔记 &#x1f308;喜欢的诗句:无人扶我青云志 我自踏雪至山巅 文章目录 **内存和地址(知识铺垫(了解即可))**如何理解编址**指针变量*…

故障诊断 | 基于LSTM的滚动轴承故障诊断

效果 概述 基于LSTM(长短期记忆网络)的滚动轴承故障诊断是一种利用深度学习技术来预测滚动轴承是否存在故障的方法。下面是一个基本的滚动轴承故障诊断的流程: 数据收集:首先,需要收集与滚动轴承相关的振动信号数据。这些数据可以通过传感器或振动监测系统获取。收集的数…

OSI七层网络模型 —— 筑梦之路

在信息技术领域&#xff0c;OSI七层模型是一个经典的网络通信框架&#xff0c;它将网络通信分为七个层次&#xff0c;每一层都有其独特的功能和作用。为了帮助记忆这七个层次&#xff0c;有一个巧妙的方法&#xff1a;将每个层次的英文单词首字母组合起来&#xff0c;形成了一句…

c# .net 香橙派 Orangepi GPIO高低电平、上升沿触发\下降沿触发 监听回调方法

c# .net 香橙派GPIO高低电平、上升沿触发\下降沿触发 监听回调方法 通过gpio readall 查看 gpio编码 这里用orangepi zero3 ,gpio= 70为例 当gpio 70 输入高电平时,触发回调 c# .net 代码 方法1: Nuget 包 System.Device.Gpio ,微软官方库对香橙派支持越来越好了,用得…