当前位置：首页 > news >正文

GPT - 因果掩码（Causal Mask）

news 来源：原创 2025/9/16 12:08:35

本节代码定义了一个函数 causal_mask，用于生成因果掩码（Causal Mask）。因果掩码通常用于自注意力机制中，以确保模型在解码时只能看到当前及之前的位置，而不能看到未来的信息。这种掩码在自然语言处理任务（如语言生成）中非常重要，因为它模拟了人类阅读或写作时的顺序性。

一、因果掩码（Causal Mask）代码实现

def causal_mask(x):mask = torch.triu(torch.ones(x.shape[0], x.shape[0]), diagonal=1) == 0return mask

1. 输入参数

x：输入张量，通常是一个序列，形状为 (seq_len, d_model) 或 (batch_size, seq_len, d_model)。这里的 seq_len 是序列的长度。

2. 生成掩码

mask = torch.triu(torch.ones(x.shape[0], x.shape[0]), diagonal=1) == 0

torch.ones(x.shape[0], x.shape[0])：生成一个形状为 (seq_len, seq_len) 的全1矩阵。
torch.triu(..., diagonal=1)：取该矩阵的上三角部分（包括对角线），其余部分设置为0。diagonal=1 表示从对角线的下一个位置开始取上三角部分。
== 0：将上三角部分（包括对角线）的值设置为 False，其余部分设置为 True。这样生成的掩码矩阵中，True 表示需要保留的注意力位置，False 表示需要被忽略的注意力位置。

3. 返回值

mask：生成的因果掩码，形状为 (seq_len, seq_len)，是一个布尔张量。

示例

假设输入张量 x 的形状为 (5, d_model)，即序列长度为5。那么：

x = torch.randn(5, d_model)  # 示例输入
mask = causal_mask(x)
print(mask)

输出的掩码矩阵 mask 将是：

tensor([[ True, False, False, False, False],[ True,  True, False, False, False],[ True,  True,  True, False, False],[ True,  True,  True,  True, False],[ True,  True,  True,  True,  True]])

作用

在自注意力机制中，因果掩码用于确保模型在计算注意力分数时，只能看到当前及之前的位置，而不能看到未来的信息。具体来说：

True：表示可以计算注意力分数。
False：表示需要被忽略，注意力分数会被设置为一个非常小的值（如 -1e9），从而在 softmax 归一化后，其权重趋近于0。

二、因果掩码如何使用？

1. 因果掩码的生成

因果掩码的生成函数如下：

def causal_mask(x):mask = torch.triu(torch.ones(x.shape[0], x.shape[0]), diagonal=1) == 0return mask

输入：x 是一个张量，通常是一个序列的嵌入表示，形状为 (seq_len, d_model) 或 (batch_size, seq_len, d_model)。
输出：生成一个布尔张量 mask，形状为 (seq_len, seq_len)，其中上三角部分（包括对角线）为 True，其余部分为 False。

2. 因果掩码的应用

因果掩码在 Poetry 数据集类中被应用，具体如下：

class Poetry(Dataset):def __init__(self, poetries, tokenizer: Tokenizer):self.poetries = poetriesself.tokenizer = tokenizerself.pad_id = self.tokenizer.vocab["[PAD]"]self.bos_id = self.tokenizer.vocab["[BOS]"]self.eos_id = self.tokenizer.vocab["[EOS]"]def __len__(self):return len(self.poetries)def __getitem__(self, idx):poetry = self.poetries[idx]poetry_ids = self.tokenizer.encode(poetry)input_ids = torch.tensor([self.bos_id] + poetry_ids)input_msk = causal_mask(input_ids)label_ids = torch.tensor(poetry_ids + [self.eos_id])return {"input_ids": input_ids,"input_msk": input_msk,"label_ids": label_ids}

__getitem__ 方法：
- 对于每首诗 poetry，将其编码为 poetry_ids。
- 在输入序列的开头添加 [BOS]（开始标记符），生成 input_ids。
- 使用 causal_mask 函数生成因果掩码 input_msk。
- 在标签序列的末尾添加 [EOS]（结束标记符），生成 label_ids。

3. 因果掩码的传递

在训练过程中，因果掩码 input_msk 会被传递给模型的自注意力层。具体如下：

for epoch in range(epochs):for batch in tqdm(trainloader, desc="Training"):batch_input_ids = batch["input_ids"]batch_input_msk = batch["input_msk"]batch_label_ids = batch["label_ids"]output = model(batch_input_ids, batch_input_msk)loss = loss_fn(output.view(-1, len(vocab)), batch_label_ids.view(-1))loss.backward()optim.step()optim.zero_grad()

model(batch_input_ids, batch_input_msk)：
- batch_input_ids 是输入序列的嵌入表示。
- batch_input_msk 是对应的因果掩码。
- 模型在计算自注意力时，会使用 batch_input_msk 来确保解码器只能看到当前及之前的位置。

4. 因果掩码的作用

在 MultiHeadAttention 类中，因果掩码被应用到注意力分数矩阵中：

if attn_mask is not None:attn_mask = attn_mask.unsqueeze(1)atten_scores = atten_scores.masked_fill(attn_mask == 0, -1e9)

attn_mask.unsqueeze(1)：
- 将掩码的形状从 (batch_size, seq_len, seq_len) 扩展为 (batch_size, 1, seq_len, seq_len)。
masked_fill：
- 将掩码中为 False 的位置的注意力分数设置为 -1e9，确保这些位置的注意力权重趋近于0。

5. 生成诗歌时的因果掩码

在生成诗歌时，因果掩码同样被应用：

def generate_poetry(method="greedy", top_k=5):model.eval()with torch.no_grad():input_ids = torch.tensor(vocab["[BOS]"]).view(1, -1)while input_ids.shape[1] < seq_len:output = model(input_ids, None)probabilities = torch.softmax(output[:, -1, :], dim=-1)if method == "greedy":next_token_id = torch.argmax(probabilities, dim=-1)elif method == "top_k":top_k_probs, top_k_indices = torch.topk(probabilities[0], top_k)next_token_id = top_k_indices[torch.multinomial(top_k_probs, 1)]if next_token_id == vocab["[EOS]"]:breakinput_ids = torch.cat([input_ids, next_token_id.view(1, 1)], dim=1)return input_ids.squeeze()

model(input_ids, None)：
- 在生成诗歌时，输入序列 input_ids 会逐渐增长，但因果掩码是隐含的，因为模型的自注意力层会自动处理序列的顺序性。
- 生成过程中，模型只能看到当前及之前的位置，这与训练时使用因果掩码的目的相同。

GPT - 因果掩码（Causal Mask）

本节代码定义了一个函数 causal_mask，用于生成因果掩码（Causal Mask）。因果掩码通常用于自注意力机制中，以确保模型在解码时只能看到当前及之前的位置，而不能看到未来的信息。这种掩码在自然语言处理任务（如…...

编程日记 2025/9/16 12:08:35

我使用lombok的Data注解带来的坑。代码如下： 公共类： package com.tyler.oshi.common;import lombok.Data; import lombok.NoArgsConstructor;/*** author: TylerZhong* description:*/Data NoArgsConstructor public class R {private int code;priv…...

编程日记 2025/9/16 4:57:13

基于Python的网络爬虫技术研究

基于Python的网络爬虫技术研究以下从多个方面为你介绍基于 Python 的网络爬虫技术： 概述网络爬虫是一种自动获取网页内容的程序，在 Python 中可以借助诸多强大的库和工具实现。网络爬虫能应用于数据采集、搜索引擎、舆情监测等众多领域。核心库 …...

编程日记 2025/9/13 11:26:57

微信小程序跳6

//金额格式化 rmoney: function(money) { return parseFloat(money).toFixed(2).toString().split().reverse().join().replace(/(\d{3})/g, $1,) .replace( /\,$/, ).split().reverse().join(); }, daysUntil: function(milliseconds) { const endDate new Date(milliseconds…...

编程日记 2025/8/26 22:14:03

项目1笔记

Data Data 是一个常用的 Lombok 注解，主要用于 Java 类中，可以自动生成以下内容： Getter（所有字段） Setter（所有非 final 字段） toString() 方法 equals() 和 hashCode() 方法无参构造函…...

编程日记 2025/9/16 1:57:32

分享：批量识别图片文字并重命名，根据图片文字内容对图片批量重命名，Python和Tesseract OCR的完成方案

一、项目背景在日常工作中，处理大量图片文件时，常常需要从图片中提取文字信息，并根据提取的文字对图片进行重命名。传统的手动操作方式效率低下且容易出错。通过OCR（光学字符识别）技术，可以自动从图片中提取文字信息，并基于提取的文字对图片进行批量重命名。 Tesserac…...

编程日记 2025/9/10 9:38:44

【安全】加密算法原理与实战

为了理解SSL/TLS原理，大家需要掌握一些加密算法的基础知识。当然，这不是为了让大家成为密码学专家，所以只需对基础的加密算法有一些了解即可。基础的加密算法主要有哈希（Hash，或称为散列）、对称加密(Symm…...

编程日记 2025/9/15 18:24:24

STM32STM8芯片擦除与读保护

连接STM单片机与断开单片机连接， 点击擦除就可以了。文件选HEX在选择Verify进行下载。...

编程日记 2025/9/3 1:35:34

Qwen2.5技术报告阅读

论文概述 ⸻ 🧠 1. 模型概述 Qwen2.5 是阿里巴巴推出的一系列大语言模型（LLMs），在预训练数据量和后训练方法上都比前一代 Qwen2 有了显著提升。 ⸻ 📈 2. 模型特点 • 预训练数据量提升：从 7 万亿…...

编程日记 2025/9/9 17:38:05

HDCP（二）

HDCP加密算法实现详解 HDCP（高带宽数字内容保护）的加密算法实现涉及对称加密、密钥派生、动态同步机制等核心环节，其设计兼顾实时性与安全性。以下从算法类型、流程实现、硬件集成等角度展开分析： 1. 加密算法类型与版本差异 •…...

编程日记 2025/9/15 4:24:59

POSIX线程（pthread）库：线程的终止与管理

在POSIX线程（pthread）库中，线程的终止和管理涉及多个关键函数。以下是关于线程终止的pthread系列函数的详细介绍： 1. pthread_exit：线程主动退出 ✨ 功能： 允许线程主动终止自身，并返回一个退出…...

编程日记 2025/9/15 3:25:26

Elasticsearch 系列专题 - 第三篇：搜索与查询

搜索是 Elasticsearch 的核心功能之一。本篇将介绍如何构建高效的查询、优化搜索结果，以及调整相关性评分，帮助你充分发挥 Elasticsearch 的搜索能力。 1. 基础查询 1.1 Match Query 与 Term Query 的区别 Match Query：用于全文搜索，会对查询词进行分词。 GET /my_index/_…...

编程日记 2025/9/8 16:39:25

【AI提示词】Emoji风格排版艺术与设计哲学

提示说明 Emoji风格排版艺术与设计哲学。提示词请使用 Emoji 风格编辑以下段落，该风格以引人入胜的标题、每个段落中包含表情符号和在末尾添加相关标签为特点。请确保保持原文的意思。使用案例（春日穿搭） 🌸 2025春季穿搭灵…...

编程日记 2025/9/9 7:49:09

C语言 ——— 认识C语言

认识 main 函数 main 函数是程序的入口，程序执行时会从 main 函数的第一行开始执行，且一个工程中 main 函数有且只有一个标准的 main 函数格式： int main() {return 0; } int 是类型，这里指的是 main 函数的返回类型 return…...

编程日记 2025/9/12 22:04:03

44、Spring Boot 详细讲义（一）

Spring Boot 详细讲义目录 Spring Boot 简介Spring Boot 快速入门Spring Boot 核心功能Spring Boot 技术栈与集成Spring Boot 高级主题Spring Boot 项目实战Spring Boot 最佳实践总结一、Spring Boot 简介 1. Spring Boot 概念和核心特点 1.1、什么是 Spring Boot&#…...

编程日记 2025/9/12 1:31:03

STM32硬件IIC+DMA驱动OLED显示——释放CPU资源，提升实时性

目录前言一、软件IIC与硬件IIC 1、软件IIC 2、硬件IIC 二、STM32CubeMX配置KEIL配置三、OLED驱动示例 1、0.96寸OLED 2、OLED驱动程序 3、运用示例 4、效果展示总结前言 0.96寸OLED屏是一个很常见的显示模块，其驱动方式在用采IIC通讯时，常用软件IIC…...

编程日记 2025/9/9 10:45:32

Android 中绕过hwbinder 实现跨模块对audio 的HAL调用

需求 Audio 模块中专门为 TV 产品添加了一些代码，需要在 hdmi 的 HAL 代码中进行调用以完成某些功能。解决方法首先将 hdmi HAL 要调用的 audio 接口函数所在的 .so 链接到最基本的 lib.primay.amlogic.so 中（其它平台上这个 .so 文件的名字也可能是…...

编程日记 2025/9/13 15:27:11

基于单片机技术的手持式酒精检测电路设计

基于STC89C52单片机的酒精检测仪设计目录基于STC89C52单片机的酒精检测仪设计一、简介二、酒精测试仪总体方案设计2.1 酒精检测仪设计要求分析2.2 设计框图三、硬件设计3.1 酒精检测电路3.2 模数转换电路3.3 STC89c52单片机电路3.4 LED显示电路3.5 声光报警电路3.6 按键和复…...

编程日记 2025/9/12 12:56:27

【车道线检测（0）】卷首语

车道线检测领域，早期的LaneNet、CondLaneNet等模型。现在在精度、实时性、复杂场景适应性等方面有了更多进展。 Head（输出头）的设计角度分类在车道线检测任务中，Head（输出头）的设计角度直接影响模型的…...

编程日记 2025/9/11 5:24:50

记一次某网络安全比赛三阶段webserver应急响应解题过程

0X01 任务说明 0X02 靶机介绍 Webserver（Web服务器）是一种软件或硬件设备，用于接收、处理并响应来自客户端（如浏览器）的HTTP请求，提供网页、图片、视频等静态或动态内容，是互联网基础设施的核心…...

编程日记 2025/9/15 1:09:44

AI 越狱技术剖析：原理、影响与防范

一、AI 越狱技术概述 AI 越狱是指通过特定技术手段，绕过人工智能模型（尤其是大型语言模型）的安全防护机制，使其生成通常被禁止的内容。这种行为类似于传统计算机系统中的“越狱”，旨在突破模型的限制，以实…...

编程日记 2025/9/10 18:52:34

项目进度延误的十大原因及应对方案

项目进度延误主要源于以下十大原因：目标不明确、需求频繁变更、资源配置不足或不合理、沟通不畅、风险管理不足、缺乏有效的项目监控、技术难题未及时解决、团队协作效率低下、决策链过长、外部因素影响。其中，需求频繁变更是导致延误的关键因素之一&…...

编程日记 2025/9/16 10:10:34

瑞友客户端登录GS_ERP时，报错：由于安全许可证服务器不能提供许可证,连接被中断的解决方法

瑞友客户端登录GS_ERP时，报错：由于安全许可证服务器不能提供许可证，连接被中断的解决方法瑞友客户端登录GS_ERP时， 报错：由于安全许可证服务器不能提供许可证，连接被中断的解决方法是由于远程桌面连接协议…...

编程日记 2025/9/6 20:29:43

android wifi通过命令行打开2.4G热点

android系统支持2G和5G，但车机系统应用只支持5G，但是需要测试2.4G的射频方法如下： 1、adb shell 进去，su 指定root权限，确保热点处于关闭状态 2、开启热点为www99999, 密码为12345678， wpa2的加密协议 cm…...

编程日记 2025/9/12 9:38:47

truncate,drop,delete分析

truncate,drop,delete对比分析特性 TRUNCATE DROP DELETE **操作对象** 表中的所有数据整个表及其所有数据表中的特定数据 **是否保留表结构** 是否是 **是否可恢复** 不可恢复不可恢复可恢复 **性能** 高高低（逐行删除） …...

编程日记 2025/9/9 11:59:50

vue+flask图书知识图谱推荐系统

文章结尾部分有CSDN官方提供的学长联系方式名片文章结尾部分有CSDN官方提供的学长联系方式名片关注B站，有好处！ 编号: F025 架构: vueflaskneo4jmysql 亮点：协同过滤推荐算法知识图谱可视化支持爬取图书数据，数据超过万条&am…...

编程日记 2025/9/11 1:15:00

什么是微前端？有什么好处？有哪一些方案？

微前端（Micro Frontends） 微前端是一种架构理念，借鉴了微服务的思想，将一个大型的前端应用拆分为多个独立、自治的子应用，每个子应用可以由不同团队、使用不同技术栈独立开发和部署，最终聚合为一个整体产品…...

编程日记 2025/9/10 22:32:46

prompts提示词经典模板

prompts.py 中的提示词模板详解文件中定义了两个核心提示词模板：REASON_PROMPT 和 RELEVANT_EXTRACTION_PROMPT。这两个模板在 DeepResearcher 的推理过程中扮演着关键角色。下面我将详细解析这两个模板的结构和功能。 REASON_PROMPT 详解 REASON_PROMPT 是用于指…...

编程日记 2025/9/9 1:32:05

pytorch软件封装

封装代码，通过传入文件名，即可输出类别信息上一章节，我们做了关于动物图像的分类，接下来我们把程序封装，然后进行预测。单张图片的predict文件 predict.py 按着路径，导入单张图片做预测from torchvis…...

编程日记 2025/9/9 7:54:02

DAY02：【pytorch】计算图与动态图机制

一、引言在深度学习框架中，计算图是理解自动求导和模型优化的核心概念。无论是PyTorch的动态图机制，还是TensorFlow早期的静态图模式，计算图都扮演着关键角色。本文将深入解析计算图的基本原理，并结合PyTorch代码演示动态图的运…...

编程日记 2025/9/12 6:52:06

基于元学习（Meta-Learning）的恶意流量检测

元学习（Meta-Learning），即“学会学习”（Learning to Learn），通过让模型从多个相关任务中提取通用知识，实现对新任务的快速适应。在恶意流量检测领域，元学习的核心价值在于从少量攻击样本中泛化出检测规则，尤其适用于新型攻击、定向APT攻击等数据稀缺场景。一、元学习…...

编程日记 2025/9/16 0:52:13

JavaScript：游戏开发的利器

在近年来的科技迅速发展中，JavaScript 已逐渐成为游戏开发领域中最受欢迎的编程语言之一。它的跨平台特性、广泛的社区支持、丰富的库和框架使得开发者能够快速、有效地创建各种类型的游戏。本文将深入探讨 JavaScript 在游戏开发中的优势。一、跨平台支持 JavaSc…...

编程日记 2025/9/11 10:09:19

玩转Docker | 使用Docker搭建pinry图片展示系统

玩转Docker | 使用Docker搭建pinry图片展示系统前言一、Pinry介绍Pinry 简介Pinry 特点二、系统要求环境要求环境检查Docker版本检查检查操作系统版本三、部署pinry服务下载镜像创建容器检查容器状态检查服务端口安全设置四、访问pinry应用五、pinry基本使用5.1 注册pinry账号…...

编程日记 2025/9/16 1:57:33

DAPP实战篇：使用web3.js实现前端输入钱包地址查询该地址的USDT余额——前端篇

专栏：区块链入门到放弃查看目录-CSDN博客文章浏览阅读385次。为了方便查看将本专栏的所有内容列出目录，按照顺序查看即可。后续也会在此规划一下后续内容，因此如果遇到不能点击的，代表还没有更新。声明：文中所出观点大多数源于笔者多年开发经验所总结，如果你想要知道区块…...

编程日记 2025/9/16 1:10:36

企业信息化-系统架构师（九十八）

1在对一个软件系统进行设计与确认后，需要进行架构复审，架构复审目的标识潜在风险，及早发现架构的缺陷和漏洞，在架构复审过程中，主要由（）决定架构满足需求，质量需求是否在设计中体现。…...

编程日记 2025/9/11 7:59:14

【C++算法】54.链表_合并 K 个升序链表

文章目录题目链接：题目描述：解法C 算法代码： 题目链接： 23. 合并 K 个升序链表题目描述： 解法解法一：暴力解法每个链表的平均长度为n，有k个链表，时间复杂度O(nk^2) 合并两个有序…...

编程日记 2025/9/7 16:32:28

阿里云CDN应对DDoS攻击策略

阿里云CDN遭遇DDoS攻击时，可通过以下综合措施进行应对，保障服务的稳定性和可用性： 1. 启用阿里云DDoS防护服务阿里云提供专业的DDoS防护服务，通过流量清洗中心过滤恶意流量，确保合法请求正常传输。该服务支持按需选…...

编程日记 2025/9/8 13:44:31

MySQL8的索引跳跃扫描原理

#MySQL 8 的索引跳跃扫描（Index Skip Scan）原理 1. 什么是索引跳跃扫描？索引跳跃扫描（Index Skip Scan）是 MySQL 8.0.13 引入的一种优化技术，允许在某些情况下跳过联合索引的最左前缀字段，仍然…...

编程日记 2025/9/8 10:27:21

centos 启动nginx 服务器

✅ 如果你是通过 yum 安装的 Nginx（推荐方式）： 🔹 启动 Nginx： sudo systemctl start nginx 🔹 设置开机自启（建议开启）： sudo systemctl enable nginx &#x1f53…...

编程日记 2025/9/14 4:00:02

格式化输出

% 符号相关数据类型代码 %s：字符串示例：print("名字是 %s" % "Tom") → 名字是 Tom%c：字符/ASCII码示例：print("%c" % 65) → A%d/%i：有符号整数示例：print("年龄…...

编程日记 2025/9/14 5:22:15

[leetcode]动态规划：斐波那契数列

一.线性dp 1.0什么是线性dp 线性DP就是指状态的转移具有线性递推关系，每个状态只依赖之前的状态，按照线性顺序一步步递推下去。 1.1斐波那契数列问题 #include <iostream> #include <vector> using namespace std; int main() { in…...

编程日记 2025/9/12 13:31:13

HackMyVM - todd记录

HackMyVM - toddhttps://mp.weixin.qq.com/s/E_-hepdfY-0veilL1fl2QA...

编程日记 2025/8/17 19:42:14

【spark认任务提交】配置优先级顺序

配置优先级顺序 Spark-submit 命令行参数 (最高优先级)代码中通过 SparkConf 设置的参数 (在应用程序中直接设置)spark-defaults.conf 文件中的配置实际应用中的建议固定配置：将集群级别的默认配置放在 spark-defaults.conf 中应用特定配置：将应用特…...

编程日记 2025/9/14 1:31:48

如何建立高效的会议机制

建立高效的会议机制需做到：明确会议目标、制定并提前分发议程、控制会议时长、确保有效沟通与反馈、及时跟进执行情况。其中，明确会议目标是核心关键，它直接决定了会议的方向与效率。只有明确目标，会议才不会偏离初衷，…...

编程日记 2025/9/15 23:14:40

spark Core-RDD转换算子

1. map算子：对RDD中的数据逐条进行映射转换，可实现类型或值的转换。函数签名为 def map[U: ClassTag](f: T > U): RDD[U] 。 2. mapPartitions算子：以分区为单位处理数据，可进行任意处理。与 map 相比， map 是分区内…...

编程日记 2025/9/15 9:39:39

【图像处理】C++实现通用Raw图转Unpack14的高效方法

一、需求背景在图像处理领域，我们经常需要处理各种位深的原始数据（如Raw8、Unpack10等）。某些高端相机或传感器会输出14位精度的图像数据，但受传输限制，实际存储时可能采用低位深打包。本文将实现一个通用转换函数&a…...

编程日记 2025/9/5 11:22:33

Vue3的Composition API与React Hooks有什么异同？

Vue3的一个重大更新点就是支持Composition API，而且也被业界称为hooks，那么Vue3的“Hooks”与React的Hooks有这么区别呢？ 一、核心相似点 1. 逻辑复用与代码组织都解决了传统类组件或选项式 API 中逻辑分散的问题，允许将相关逻…...

编程日记 2025/9/14 6:15:12

Gerrit的安装与使用说明（Ubuntu）

#本页面按192.168.60.148服务器举例进行安装配置 1.权限配置 ## 使用root或者有sudo权限用户执行 # 创建gerrit用户 sudo useradd gerrit # 设置gerrit用户的密码 sudo passwd gerrit # 增加sudo权限 sudo visudo 在root ALL(ALL:ALL) ALL行下添加如下内容 gerrit ALL(ALL:…...

编程日记 2025/9/15 16:45:59

如何在Git历史中抹掉中文信息并翻译成英文

如何在Git历史中抹掉中文信息并翻译成英文在软件开发和版本控制领域，维护一个清晰、一致的代码历史记录是至关重要的。然而，有时我们可能会遇到需要修改历史提交的情况，比如删除敏感信息或修正错误。本文将详细探讨如何在Git历史中抹掉中文…...

编程日记 2025/9/11 17:26:09

Ubuntu利用docker将ONNX模型转换为RK3588模型

1.安装docker 下载教程 1.拉取镜像方法一：通过命令拉取 # 下载官方Docker镜像sudo docker pull registry.cn-hangzhou.aliyuncs.com/rockchip/rknn-toolkit2:v2.3.0 方法二：通过rknn-toolkit2自带的直接安装 2.开始工作创建工作目录并复制ONNX模型…...

编程日记 2025/9/7 1:26:05