1. 环境准备:从驱动检查到依赖安装
第一次在Linux上装CUDA时,我盯着命令行界面足足发了五分钟呆——NVIDIA驱动、CUDA Toolkit、环境变量这些概念像一团乱麻。后来才发现,安装CUDA本质上是在搭建三层结构:底层显卡驱动是地基,中间层CUDA Toolkit是工具库,上层环境变量则是通行证。这里以Ubuntu 22.04为例,带大家走通全流程。
先检查显卡驱动是否就位。打开终端输入:
bash复制nvidia-smi
如果看到类似下面的输出,说明驱动已安装:
code复制+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.54.03 Driver Version: 535.54.03 CUDA Version: 12.2 |
|-----------------------------------------+----------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+======================+======================|
| 0 NVIDIA GeForce RTX 3090 Off | 00000000:01:00.0 Off | N/A |
| 30% 45C P8
