cuda和tensorflow對應關係

cuda和tensorflow對應關係

linux:\color{#9400D3}{linux:}

  • 查看cuda版本:

    cat /usr/local/cuda/version.txt

  • 查看cudnn版本:

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

windows:\color{#9400D3}{windows:}

  • 查看cuda版本:

    nvcc -V

  • 查看cudnn版本:

    打開cudnn文件夾 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include ,打開cudnn.h文件,我使用notepad++打開
    ctrl+F 檢索 CUDNN_MAJOR ,結果如下:
    在這裏插入圖片描述
    可判斷我的cudnn版本爲7.1.3

:\color{#9400D3}{查看版本對應關係:}

https://tensorflow.google.cn/install/source_windows

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章