使用nvidia-smi nvidia-smi是NVIDIA提供的一个用于管理和监控GPU的命令行工具。它可以显示驱动程序版本和CUDA版本。

nvidia-smi

输出示例：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 555.58.02    Driver Version: 555.58.02    CUDA Version: 12.5     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Tesla K80           On   | 00000000:00:1E.0 Off |                    0 |
| N/A   39C    P8    29W / 149W |      0MiB / 11441MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+