1. 查看cuda版本
nvcc -V
或:cat /usr/local/cuda/version.txt
2. 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
或:cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2
- Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
- windows
https://tensorflow.google.cn/install/source_windows - Linux
https://tensorflow.google.cn/install/source
未完待续
碰到的几种坑:
- 问题: tensorflow的CUDA driver version is insufficient for CUDA runtime
https://blog.csdn.net/li57681522/article/details/82491617 - 问题:could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
https://blog.csdn.net/xianqin_ma/article/details/79525519
cuda版本升级:
https://blog.csdn.net/wanzhen4330/article/details/81704474