美文网首页
006 CUDA/CUDNN与tensorflow

006 CUDA/CUDNN与tensorflow

作者: Oscar_hailiang | 来源:发表于2019-04-14 14:53 被阅读0次
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
  1. Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
碰到的几种坑:
  1. 问题: tensorflow的CUDA driver version is insufficient for CUDA runtime
    https://blog.csdn.net/li57681522/article/details/82491617
  2. 问题: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