Windows11安装GPU版本Pytorch2.6教程

news/2025/2/24 18:40:09

1: 准备工作

针对已经安装好的Windows11系统,先检查Nvidia驱动和使用的CUDA版本情况。先打开Windows PowerShell,通过nvidia-smi命令查看GPU的情况,结果如下图1所示,从结果中可知使用的CUDA版本为12.8。

图1:检测安装好的CUDA版本情况

图1:检测安装好的CUDA版本情况

然后,通过Pytorch官网(https://pytorch.org/),查看目前Pytorch的稳定版本支持的CUDA版本,从下图2中可以发现最高支持的CUDA是12.6。因此需要降低已经安装Nvidia驱动的CUDA版本。

图2:Pytorch和CUDA的匹配情况

图2:Pytorch和CUDA的匹配情况

我的方法是,重新安装包含CUDA12.6的NVIDIA Studio驱动。CUDA版本和驱动的匹配情况可以通过下面的链接查看。从图3的结果中可以知道,可以选择560、561等版本的驱动。https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

图3:CUDA版本和驱动的匹配情况

图3:CUDA版本和驱动的匹配情况

可以通过直接搜索对应的版本,然后直接下载,对应的版本情况如下图4所示:

图4:版本560.94的Nvidia驱动

图4:版本560.94的Nvidia驱动

也可以通过https://www.nvidia.cn/geforce/drivers/网站搜索可以使用的驱动并下载,图5展示的是搜索到的可用的Nvidia Studio驱动(深度学习计算建议安装Nvidia Studio类型的驱动)。这次选择的驱动版本为560.81。

图5:搜索到的Nvidia Studio驱动

图5:搜索到的Nvidia Studio驱动

下载560.81版本的Nvidia Studio后,直接安装即可,Nvidia会自动对已经安装的驱动进行必要的替换。安装完成后,重新使用nvidia-smi后可见CUDA的版本已经为12.6。如图6所示。

图6:查看新的CUDA版本

图6:查看新的CUDA版本

2: 安装Python

Python的安装可以通过Anaconda来完成,Anaconda的下载链接如下:https://www.anaconda.com/download/success,下载界面如下图7所示,选择下载Windows版本对应的安装程序。

图7:Anaconda下载界面

图7:Anaconda下载界面

安装Anaconda完成后,打开应用后的界面如下图8所示:

图8:Anaconda应用界面

图8:Anaconda应用界面

在安装好的Python应用中,针对Pytorch创建一个新的环境,用于安装Pytorch相关的库。图9展示的是创建了名字为Pytorch26的新环境,用于后续安装Pytorch2.6版本。

图9:在Anaconda中创建新的Python环境

图9:在Anaconda中创建新的Python环境

3: 安装GPU版本的Pytorch

通过安装Anaconda时同步安装的Anaconda Prompt,来安装GPU版本的Pytorch。需要先通过activate Pytorch26命令激活新建的Pytorch26。然后通过pip相关的命令,进行Pytorch安装。安装命令可以通过图10中直接复制。

图10:获取安装Pytorch的命令行

图10:获取安装Pytorch的命令行

安装Pytorch时在Anaconda Prompt需要运行的命令行下如图11所示,图12展示了已经成功安装的相关库。

图11:在Anaconda Prompt需要运行的命令

图11:在Anaconda Prompt需要运行的命令

图12:已经成功安装的相关库

图12:已经成功安装的相关库

在安装成功后,可以打开Python的JupyterLab应用,来查看Pytorch26环境中是否已经成功安装GPU版本的Pytorch。命令和对应的结果如图13所示,可以发现,已经成功安装了GPU版本的Pytorch2.6。

图13:检查是否成功安装GPU版本的Pytorch

图13:检查是否成功安装GPU版本的Pytorch


http://www.niftyadmin.cn/n/5864707.html

相关文章

LLM2CLIP论文学习笔记:强大的语言模型解锁更丰富的视觉表征

1. 写在前面 今天分享的一篇论文《LLM2CLIP: P OWERFUL L ANGUAGE M ODEL U NLOCKS R ICHER V ISUAL R EPRESENTATION》, 2024年9月微软和同济大学的一篇paper, 是多模态领域的一篇工作,主要探索了如何将大模型融合到Clip模型里面来进一步提…

计算机视觉行业洞察--影像行业系列第一期

计算机视觉行业产业链的上下游构成相对清晰,从基础技术研发到具体应用场景的多个环节相对成熟。 以下是我结合VisionChina经历和行业龙头企业对计算机视觉行业产业链上下游的拆解总结。 上下游总结 上游产业链分为软硬件两类,视觉的硬件主要指芯片、…

网络安全防护指南:筑牢网络安全防线(510)

一、网络安全的基本概念 (一)网络的定义 网络是指由计算机或者其他信息终端及相关设备组成的按照一定的规则和程序对信息收集、存储、传输、交换、处理的系统。在当今数字化时代,网络已经成为人们生活和工作中不可或缺的一部分。它连接了世…

(四)趣学设计模式 之 原型模式!

目录 一、 啥是原型模式?二、 为什么要用原型模式?三、 原型模式怎么实现?四、 原型模式的应用场景五、 原型模式的优点和缺点六、 总结 🌟我的其他文章也讲解的比较有趣😁,如果喜欢博主的讲解方式&#xf…

【单片机毕业设计14-基于stm32c8t6的智能宠物养护舱系统设计】

【单片机毕业设计14-基于stm32c8t6的智能宠物养护舱系统设计】 前言一、功能介绍二、硬件部分三、软件部分总结 前言 🔥这里是小殷学长,单片机毕业设计篇14-基于stm32c8t6的智能宠物养护舱系统设计 🧿创作不易,拒绝白嫖可私 一、功…

vue怎么设置允许局域网手机访问

打开vite.config.ts 添加 server: {host: 0.0.0.0}, host: 0.0.0.0:设置为0.0.0.0,允许从所有IP访问。port: 5173:指定端口号,可以根据需要进行修改。不指定默认 5173disableHostCheck: true:禁用主机检查&#xff0c…

C#贪心算法

贪心算法:生活与代码中的 “最优选择大师” 在生活里,我们常常面临各种选择,都希望能做出最有利的决策。比如在超市大促销时,面对琳琅满目的商品,你总想用有限的预算买到价值最高的东西。贪心算法,就像是一…

Android KMP初探

Android KMP初探 前言: 最近线上听了Kotlin官网举行的KMP会议,感觉听神奇的,于是就把官方demo下载下来尝试了一下,下载插件和所需要的依赖都用了很久,但是发现里面的代码很少,于是尝试自己手写了一下&…