发布时间:2024-10-17 11:01
最近在装pytorch的时候,看到了一个选择CUDA版本的选项
之前从来没用过CUDA这玩意,当然也不知道CUDA版本啥玩意,于是找了与一下
CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
关于查看自己电脑上的CUDA版本,首先需要区分CUDA的两种API,即运行时API(Runtime API)和驱动API(Driver API)
驱动API(Driver API)由GPU driver installer安装
在搜索结果中选NVIDIA Control Panel
双击打开后如图所示
命令行输入nvidia-smi
右上角的CUDA Version便是CUDA Driver API版本
查看运行时API 版本的前提是已将在电脑上安装了CUDA Toolkit
命令行输入nvcc -V
最下边一行便是CUDA Runtime API的版本
注意:Driver API和Runtime API可以不同,具体应用要看使用场景
有关二者的不同可以参考这篇文章https://blog.csdn.net/weixin_39518984/article/details/111406728