首先明确四个概念:
1. Cuda本身拥有两个API,分别是cuda driver version 和 cuda runtime version。
2. Win + R 打开命令行窗口,执行nvidia-smi查看的是cuda driver version版本(如图1)。cuda driver version是跟nvidia的GPU驱动绑定在一起的(通俗来讲就是与你的电脑显卡绑定在一起的),已经固定了的,是改变不了的。
3. Win + R 打开命令行窗口,执行nvcc -V(或nvcc --version)查看的是cuda runtime version版本(前提是你正确安装了cuda并将cuda加入了环境变量,如图2),CUDA Runtime Version是你自己在系统上安装的cuda版本,也是你跑深度学习模型或其它程序调用的cuda版本。
4. CUDA Driver Version和CUDA Runtime Version的版本不必非要一致,但CUDA Runtime Version要<=CUDA Driver Version。

图1

图2 

综上所述:输入nvidia-smi查看cuda driver version,版本为11.7,说明本机cuda的绑定的驱动版本为11.7,所以可以支持<=11.7版本的cuda安装,于是安装cuda时就选择<=11.7的版本安装。安装好cuda后输入nvcc –V查看cuda runtime version版本,版本为10.1(因为我安装的版本是10.1的,10.1<=11.7符合要求)。

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐