Ubuntu 23.04 验证 GPU 驱动是否正常安装

通过 watch -n 1 nvidia-smi 命令查看GPU的运行状态

通过 torch 命令查看

查看显卡数量，

import torch
torch.cuda.device_count()

如果只有 1 张显卡，输出结果应该是 1。

验证 cuda 是否有效，有效的话，输出结果应该是 True。

import torch
torch.cuda.is_available()

完结！