Windows11安装GPU版本Pytorch2.6教程

1: 准备工作

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

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

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

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

图4:版本560.94的Nvidia驱动

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

图5:搜索到的Nvidia Studio驱动

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

图6:查看新的CUDA版本

2: 安装Python

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

图7:Anaconda下载界面

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

图8:Anaconda应用界面

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

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

3: 安装GPU版本的Pytorch

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

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

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

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

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

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

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

相关推荐
中年程序员一枚8 分钟前
让postman调用python的开发接口,链接sqlite数据库,让前后联动起来
数据库·python·postman
火火PM打怪中10 分钟前
空窗期的自我探索
人工智能·职场和发展
测试人社区-千羽12 分钟前
AI重塑API测试数据生成的时代背景
人工智能·测试工具·程序人生·自动化·测试覆盖率
爱写Bug的小孙18 分钟前
Tools、MCP 和 Function Calling
开发语言·人工智能·python·ai·ai编程·工具调用
rgb2gray20 分钟前
城市韧性与交通基础设施系统耦合协调度的时空演变及影响因素
网络·人工智能·python·ai·写作·耦合·耦合协调
初级炼丹师(爱说实话版)26 分钟前
大模型部署-数据并行/模型并行
人工智能·python
listhi52029 分钟前
机械系统运动学与动力学在MATLAB及SimMechanics中的实现方案
人工智能·算法·matlab
AI大模型学徒29 分钟前
大模型应用开发(十五)_知识库1
人工智能·chatgpt·大模型·llm·知识库·deepseek
音视频牛哥31 分钟前
从“十五五”到数字化转型:音视频技术在未来产业中的关键作用
人工智能·深度学习·计算机视觉·音视频·十五五规划音视频低延迟方案·十五五规划低空经济低延迟方案·rtsp rtmp播放器
测试人社区—小叶子38 分钟前
测试开发面试高频“灵魂八问”深度解析与应答策略
网络·人工智能·测试工具·云原生·容器·面试·职场和发展