RTX5060 Ti显卡安装cuda版本PyTorch踩坑记录

RTX5060 Ti显卡安装cuda版本PyTorch踩坑记录

问题如下:

bash 复制代码
NVIDIA GeForce RTX 5060 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 compute_37.
If you want to use the NVIDIA GeForce RTX 5060 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

这个警告的核心问题是:你安装的 PyTorch 版本过旧,其支持的 CUDA 计算能力(最高 sm_86)无法匹配 NVIDIA GeForce RTX 5060 Ti 的计算能力(sm_120,属于 NVIDIA 最新的 Blackwell 架构)。只有安装适配 Blackwell 架构的 PyTorch 版本,才能消除警告并让 GPU 正常发挥性能。

一、关键背景:GPU 计算能力与 PyTorch 适配

RTX 5060 Ti:属于 Blackwell 架构,计算能力为 sm_120,是 2024 年后推出的新显卡,需要支持该架构的 PyTorch 版本(通常是 PyTorch 2.3 及以上,且依赖 CUDA 12.4/12.5)。

当前 PyTorch:仅支持到 sm_86(Ampere 架构,如 RTX 30/40 系列),完全不兼容 sm_120,因此会触发兼容性警告,甚至可能导致 GPU 无法被调用(仅用 CPU 运行)。

我的解决方案:

1.我先是按照这个博客进行操作的,https://blog.csdn.net/qq_67105081/article/details/137519207

使用nvidia-smi 在cmd上:我的cuda是cuda version 13.0

2.我后续装安装cuda及cudnn是装的12.8

cuda链接:https://developer.nvidia.com/cuda-toolkit-archive

然后去cudnn:https://developer.nvidia.com/rdp/cudnn-archive下载cudnn

3.装好cudnn和cuda后显示使用 nvcc -V指令后:

4.在装annaconda和pycharm后,创建conda环境

bash 复制代码
conda create -n yolov12 python=3.9

5.最关键的一步来了,去pytorch官网(https://pytorch.org/)
找到这个网址(https://download.pytorch.org/whl/cu128)进去:

然后找到torch进去:

找到这个版本 https://download.pytorch.org/whl/torch/torch-2.8.0+cu128-cp39-cp39-win_amd64.whl进行下载。

在这个yolov12的conda环境下进行安装运行下面指令:

bash 复制代码
pip install  torch-2.8.0+cu128-cp39-cp39-win_amd64.whl #(我是绝对路径)

6.最后运行这个指令:

bash 复制代码
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128

按照上述安排我的问题就可以匹配解决了。

我主要是参考了这个博主,非常感谢他。
https://blog.csdn.net/xiaoyanboke/article/details/149910509?spm=1001.2101.3001.6650.2&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ECtr-2-149910509-blog-151177663.235%5Ev43%5Epc_blog_bottom_relevance_base9&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ECtr-2-149910509-blog-151177663.235%5Ev43%5Epc_blog_bottom_relevance_base9&utm_relevant_index=4

相关推荐
jinggongszh14 分钟前
使用AI协同开发产品条码规则功能——从“理解方案、原型、接口文档”到“落地真实前端代码”
前端·人工智能·mes系统·mes工程架构
专注仿真18 分钟前
问答大模型技术方案算法实现-RAPTOR树构建算法与BEG集成使用
python·算法
宋哥转AI22 分钟前
深入理解 AI Agent · MEMORY #02:记忆的工程机制
人工智能·agent·ai编程
一名普通的电源工程师30 分钟前
WRF2424S‑3WR2 适配优选 钡特电源 VF3‑24S24S|3W 工业 DC‑DC 模块电源选型拆解
人工智能·电源模块·工业电源
野三关彭于晏31 分钟前
Agent 时代,如何用 AI 重塑教育信息化
人工智能·agent
哔哩哔哩技术38 分钟前
IndexTTS 2.5 让声音跨越语言
人工智能
Mr_凌宇1 小时前
codex开启1M上下文
人工智能·程序员
AI英德西牛仔1 小时前
豆包导出 pdf 颜色不一样怎么办,选用 AI 导出鸭优化文档导出,结合行业白皮书数据解析色彩失真成因
人工智能·ai·chatgpt·pdf·deepseek·ai导出鸭
神奇小汤圆1 小时前
RAG检索优化实战:从67%到92%,我做了这4步
人工智能
老司机张师傅1 小时前
让 AI 之间自动对话:基于 TRAE Solo 的多智能体团队协作系统设计与实现
人工智能·aiaitrae