ubuntu 之 查看 cuda,cudnn版本
生活随笔
收集整理的這篇文章主要介紹了
ubuntu 之 查看 cuda,cudnn版本
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
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 22.查看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? ? ? ? ? ? ? ? ? 8cudnn 和 cuda 版本對應關系:
cuDNN Archive | NVIDIA Developer
與50位技術專家面對面20年技術見證,附贈技術全景圖總結
以上是生活随笔為你收集整理的ubuntu 之 查看 cuda,cudnn版本的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 巨坑 之 pip install 和 c
- 下一篇: pip 指定路径安装工具包