Windows 下 PyTorch 的安装(GPU版)

第一步:找到匹配表

PyTorch + CUDA + cuDNN + Python 匹配表

PyTorch 版本 支持的 CUDA 版本 推荐 cuDNN 版本(对应 CUDA) 支持的 Python 版本 备注
2.5.x 12.1、12.4 cuDNN 8.9+(12.1)、8.10+(12.4) 3.8~3.12 2025 年最新稳定版
2.4.x 12.1、12.4 cuDNN 8.9+(12.1)、8.10+(12.4) 3.8~3.12
2.3.x 11.8、12.1 cuDNN 8.7+(11.8)、8.9+(12.1) 3.8~3.11
2.2.x 11.8、12.1 cuDNN 8.7+(11.8)、8.9+(12.1) 3.8~3.11
2.1.x 11.8、12.1 cuDNN 8.7+(11.8)、8.9+(12.1) 3.8~3.11 长期支持(LTS)版本
2.0.x 11.7、11.8 cuDNN 8.5+(11.7)、8.7+(11.8) 3.8~3.11
1.13.x 11.6、11.7 cuDNN 8.4+(11.6)、8.5+(11.7) 3.7~3.10
1.12.x 11.3、11.6 cuDNN 8.2+(11.3)、8.4+(11.6) 3.7~3.10
1.11.x 11.3、10.2 cuDNN 8.2+(11.3)、8.2+(10.2) 3.7~3.9 最后支持 CUDA 10.2 的版本
1.10.x 11.3、10.2 cuDNN 8.2+(11.3)、8.2+(10.2) 3.6~3.9
1.9.x 11.1、10.2 cuDNN 8.0+(11.1)、8.0+(10.2) 3.6~3.9

第二步:查看当前nvidia驱动支持的cuda最高版本

右上角的CUDA Version是当前nvidia驱动支持的CUDA最高版本,只需要保证官网显示的PyTorch CUDA版本低于此版本即可,不用处理计算机环境本身的CUDA版本(即本机nvcc版本)我只要确保当前驱动支持的CUDA版本大于官网显示的,就可以吗,系统级的cuda版本不用管?

第三步:使用官网命令安装PyTorch

Previous PyTorch Versions

ctrl+f搜索“CUDA 12.4”:

执行命令安装

作者: 高志远

高志远,24岁,男生

发表评论

邮箱地址不会被公开。