关于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 最佳支持
相关推荐
承渊政道22 分钟前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(开源大模型ChatGLM高级定制化应用实战)
人工智能·pytorch·python·llm·模型微调·glm
Joy T2 小时前
Agent 确定性上下文设计:从 Prompt 参数注入到能力分层
java·开发语言·微服务·架构·prompt·agent context
专注仿真5 小时前
引擎扩展性与集成设计深度剖析:从 14 个扩展点到 DIS→HLA 演进之路
开发语言·算法·仿真引擎·推演引擎
Best-Wishes6 小时前
科来网络分析系统(技术交流版)使用说明书
开发语言·junit·php
代码不停6 小时前
自动化测试(java)
java·开发语言·selenium·自动化
Smoothcloud润云7 小时前
PyTorch大模型权重加载与启动耗时优化实战:定位磁盘、内存与显存瓶颈
人工智能·pytorch·python·深度学习·gpu算力
databook7 小时前
从循环到向量化:NumPy提速的7个关键思维
python·数据分析·numpy
棉猴7 小时前
玩游戏学Python1-Python和pgzero库的安装
python·pygame·玩游戏·pgzero·pgzrun
诺伦8 小时前
GEO优化数据实证:78%引用流向官网的机制拆解与小品牌三步管线实战教程 | RiseClaw玄策
python·seo·geo·内容营销·ai搜索
oooost8 小时前
pytorch学习笔记1
python·学习方法