查看cuda,cudnn版本及是否与tensorflow gpu兼容
查看cuda:nvcc --version
查看cudnn:
1、打开
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\ cudnn\include
中的cudnn.h
2、程序开头部分有
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
可以计算我的版本就是7.6.5
查看cuda,cudnn,tf gpu是否兼容
https://docs.nvidia.com/deeplearning/frameworks/tensorflow-release-notes/rel_20-02.html#rel_20-02
查看cuda,cudnn版本及是否与tensorflow gpu兼容
最新推荐文章于 2024-05-08 10:02:53 发布