报错原因：

1.pytorch版本与cuda版本不一致

2.pytorch版本不支持该显卡架构

真实事例：

博主在使用5060ti显卡运行模型时报了该错误，当时pytorch版本是cu126（最高支持sm_90） 可能原因就是cu126不支持5060ti的显卡架构（5060ti显卡架构未公布）

后来博主安装cu128的pytorch之后 报错解决。

解决办法：

1.检查pytorch版本和cuda支持的显卡架构

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.get_arch_list())  # 查看支持的架构列表

2.检查cuda版本

nvcc -V

3.查询自己的显卡架构（如H100是sm_90、RTX30系列是sm_86）

4.若cuda版本和pytorch版本不匹配则卸载原cuda或pytorch再安装cuda或pytorch使得 cuda和pytorch版本匹配

若pytorch版本不支持该显卡架构则卸载pytorch 安装支持该显卡架构的pytorch

# vscode # jupyter # pytorch
Logo
腾讯云开发者社区

腾讯云面向开发者汇聚海量精品云计算使用和开发经验，营造开放的云计算技术生态圈。

更多推荐

Elasticsearch复杂数据类型终极指南：从入门到精通

Elasticsearch作为功能强大的搜索引擎，支持多种复杂数据类型，让开发者能够灵活处理各种结构化和非结构化数据。本文将带你全面了解Elasticsearch中的复杂数据类型，从基础概念到实际应用，助你轻松掌握数据建模的核心技巧。## 内部对象：构建层级化数据结构在Elasticsearch中，对象类型（Object）是最基础的复杂数据类型之一，用于表示具有嵌套关系的数据。例如，我们可

avatar 腾讯云开发者社区

终极指南：Flink SQL连接器版本管理从混乱到有序的升级之路

Apache Flink作为流处理领域的佼佼者，其SQL连接器的版本管理一直是开发者面临的核心挑战。本文将系统讲解Flink SQL连接器版本管理的最佳实践，帮助你轻松应对版本兼容性问题，实现从混乱到有序的升级之旅。## 连接器版本管理的常见痛点 😫在Flink应用开发中，连接器版本管理常常让开发者头疼不已。不同版本的连接器可能导致各种兼容性问题，例如API变更、功能差异甚至运行时错误。

avatar 腾讯云开发者社区

如何快速搭建Neon无服务器PostgreSQL：面向初学者的完整指南

Neon是一款革命性的无服务器PostgreSQL解决方案，它通过分离存储和计算层，实现了自动扩缩容、类代码式数据库分支以及零级扩展能力。本指南将帮助你从零开始搭建Neon开发环境，体验这款创新数据库的强大功能。## 准备工作：环境要求与依赖项在开始搭建Neon环境前，请确保你的系统满足以下要求：- Linux操作系统（推荐Ubuntu 20.04+或Debian 11+）- Git

avatar 腾讯云开发者社区