ubuntu 之 查看 cuda,cudnn版本 您所在的位置:网站首页 白衣古语意思怎么说 ubuntu 之 查看 cuda,cudnn版本

ubuntu 之 查看 cuda,cudnn版本

2023-07-21 20:51| 来源: 网络整理| 查看: 265

1.查看cudnn版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

新版本变化:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

2.查看cuda版本则有很多种办法:

①:

nvidia-smi

②:

nvcc -V

③:

cat /usr/local/cuda/version.txt

细心地同学会发现,有时①查看的cuda版本和②、③查出来的会不同,这是什么原因呢。原来cuda的API分为两种类型,一种是驱动(driver)API,另一种是运行(runtime)API。而我们通过nvidia-smi查看的cuda版本正是驱动API版本。通过nvcc -V 和 cat /usr/local/cuda/version.txt查看的一般是运行API,因此两者存在出入是正常的。

但是我们运行程序时所使用的是运行API,我们时常讲的cuda,cudnn,tensorflow-gpu版本之间的对应,也是针对运行API来讲的,也就是后两者查看的版本。

附:对应关系:

可以通过该网址查看:从源代码构建  |  TensorFlow

下面添加一条比较老的版本的配置:

tensorflow_gpu-1.0.0             2.7、3.3-3.6     --- ---       5.0                  8

cudnn 和 cuda 版本对应关系:

cuDNN Archive | NVIDIA Developer



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有