美文网首页
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