美文网首页
Ubuntu18.04LTS安装CUDNN

Ubuntu18.04LTS安装CUDNN

作者: glRu | 来源:发表于2020-07-22 14:40 被阅读0次

    cuda与cudnn需要满足关系:

    https://developer.nvidia.com/rdp/cudnn-archive

    cuda历史版本下载连接:

    https://developer.nvidia.com/cuda-toolkit-archive

    Tensorflow不同版本与CUDA及CUDNN版本对应关系:  

    https://blog.csdn.net/u013066730/article/details/88845083

    1 下载cuDNN v7.6.1 (June 24, 2019), for CUDA 10.1

    https://developer.nvidia.com/rdp/cudnn-archive

    下载完成后解压并进入文件夹:

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

    sudo chmod a+r /usr/local/cuda/include/cudnn.h

    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

    若是在无界面环境下,请下载  cuDNN Library for Linux

    查看cudnn版本

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

    从上到下输出7 6 4 即表示cudnn7.6.4

    相关文章

      网友评论

          本文标题:Ubuntu18.04LTS安装CUDNN

          本文链接:https://www.haomeiwen.com/subject/ieazkktx.html