006 CUDA/CUDNN与tensorflow
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版本对应关系
碰到的几种坑:
- 问题: 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
本文标题:006 CUDA/CUDNN与tensorflow
本文链接:https://www.haomeiwen.com/subject/gkajwqtx.html
网友评论