简介
CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台。有了cuda，可以让 pytorch 等框架直接在 gpu 上运行，速度当然更快。如果深度学习框架是车，那么 cuda 就是变速箱，可进行硬件加速，让深度学习计算任务更快。安装地址
https://developer.nvidia.com/cuda-downloads
查看显卡 cuda 核心数量
1. cuda demo
可以使用 cuda 工具查询核心数，比如我的 4070Ti 拥有 7680 个 cuda 核心。
2. 可以通过 NVIDIA 控制面板查询系统信息
查看 cuda 版本
1. nvcc -v
可以获得准确版本
2. nvidia-smi 命令
可以获得大版本
参考文档
若有收获，就点个赞吧
0 人点赞
