Skip to content

计算机系统基础 ​

AI Infra 方向(方向一、二、四)研究的是“让模型在真实系统上跑得又快又省”,绕不开计算机系统的知识:程序在 CPU 和 GPU 上怎么执行、内存和显存怎么管理、进程和容器怎么调度、数据怎么在网络里传输。

人工智能、大数据专业的课程表里这些课往往较少或较晚。大一大二不需要一次学完,先建立概念,做研究遇到具体问题时再针对性深入。

学什么、学到什么程度 ​

主题大一大二先掌握做研究时再深入资料
计算机组成与体系结构存储层次(寄存器、缓存、内存、磁盘)的速度差距;CPU 和 GPU 的区别GPU 架构、内存带宽与算力的关系(Roofline 模型)CS:APP、CMU 15-213
操作系统进程与线程、内存管理与虚拟内存、文件系统的基本概念调度算法、分页(vLLM 的 PagedAttention 就借鉴了它)、容器原理(namespace、cgroup)MIT 6.1810、南京大学操作系统课
计算机网络TCP/IP 分层、HTTP、延迟与带宽的区别RPC、数据中心网络、RDMA《计算机网络:自顶向下方法》、Stanford CS144
并行计算并行与并发的区别、加速比、为什么 GPU 适合矩阵运算CUDA 编程、通信集合操作(AllReduce)Stanford CS149、PMPP
分布式系统为什么需要分布式、一致性和容错的基本概念分布式训练与推理的并行策略MIT 6.5840

推荐资料 ​

资料说明
《深入理解计算机系统》(CS:APP)与 CMU 15-213从程序员视角讲计算机系统,系统方向的“必读书”。实验(如 Data Lab、Cache Lab)质量很高
MIT 6.1810(原 6.S081)操作系统课,在教学操作系统 xv6 上动手实现系统调用、页表、调度等
南京大学操作系统:设计与实现(蒋炎岩)中文、偏动手的操作系统课,B 站有录像,看最新学期的版本
Stanford CS144计算机网络,实验中自己实现 TCP
Stanford CS149并行计算:多核、SIMD、GPU、CUDA
MIT 6.5840(原 6.824)分布式系统,实验中实现 Raft 共识算法
计算机自学指南以上课程都能在这里找到详细的学习指引和资源汇总

这些课程难度不低,不需要全部学完。建议的顺序:先读 CS:APP 的前几章和存储层次一章,再按方向选一门:做调度和容器的选操作系统,做云边协同的选网络,做推理加速的选并行计算。

和课题组研究的联系 ​

系统知识在研究中的体现
存储层次、内存带宽大模型解码阶段受显存带宽限制,KV Cache 的放置和卸载
虚拟内存与分页vLLM 的 PagedAttention 管理 KV Cache
进程调度推理请求调度、多模型共享 GPU
容器(namespace、cgroup、镜像分层)容器调度、镜像层感知的部署与迁移(见 Docker 的镜像层)
网络延迟与带宽云边协同时哪些计算放在端侧、哪些放在云端
并行与通信张量并行、流水线并行、预填充与解码分离

做 L6 Kubernetes 与调度、L8 Jetson 端侧推理、L9 层感知容器调度仿真 时,会直接用到这些知识。

作者:崔涵帅、姚智、王振胜、唐志清 · 边缘智能系统实验室