关于python版本,显卡版本,torch版本之间的问题

第一部分:显卡架构(GPU Compute Capability)对照表

显卡系列 架构 Compute Capability 是否被 PyTorch 支持
GTX 10 系 Pascal sm_61 ✔ 全部支持
RTX 20 系 Turing sm_75 ✔ 全部支持
RTX 30 系 Ampere sm_80 / sm_86 ✔ 全部支持
RTX 40 系 Ada sm_89 ✔ PyTorch ≥ 2.0 支持
RTX 50 系(5080/5090) Blackwell sm_120 ✔ PyTorch ≥ 2.5 才支持
Tesla A100 Ampere sm_80
H100 Hopper sm_90
H200 / GB200 Blackwell sm_120 / sm_130 ✔ PyTorch ≥ 2.6

第二部分:Python ↔ PyTorch 支持矩阵(官方)

PyTorch 版本 支持的 Python 支持的 CUDA RTX 5080 支持?
1.10 3.6--3.9 10.2 / 11.3 ❌ 完全不支持
1.12 3.7--3.10 11.3 / 11.6 ❌ 不支持
2.0 3.8--3.11 11.7 / 11.8 ❌ 不支持
2.1 3.8--3.11 12.1 ❌ 不支持
2.2 3.8--3.12 12.1 ❌ 不支持
2.3 3.8--3.12 12.1 ❌ 不支持
2.4 3.8--3.12 12.1 ❌ 不支持
2.5 3.10--3.12 12.4 ✔ 实验性支持 SM120
2.6 3.10--3.12 12.5 ✔ 稳定支持 SM120
2.7 3.10--3.12 12.6 ✔ 完整支持 SM120
Nightly (2.8+) 3.11--3.12 12.7 / 12.8 ✔ Blackwell 最佳支持
相关推荐
zyq99101_138 分钟前
优化二分查找:前缀和降复杂度
数据结构·python·蓝桥杯
qyzm41 分钟前
天梯赛练习(3月13日)
开发语言·数据结构·python·算法·贪心算法
leluckys1 小时前
swift- Swift中常见的面试题
开发语言·汇编·swift
BUG_MeDe1 小时前
json格式字符串解析的简单使用 libjson-c
c语言·开发语言·json
CoderCodingNo2 小时前
【GESP】C++五级练习题 luogu-P1182 数列分段 Section II
开发语言·c++·算法
Qt学视觉2 小时前
AI2-Paddle环境搭建
c++·人工智能·python·opencv·paddle
廋到被风吹走2 小时前
【LangChain4j】特点功能及使用场景
后端·python·flask
Eward-an2 小时前
LeetCode 239. 滑动窗口最大值(详细技术解析)
python·算法·leetcode