cuda和cudnn安裝

CUDA

CUDA 的安裝

  • 去英偉達官網尋找與 NVIDIA 驅動相對應的 CUDA 版本

  • sudo 運行 run 文件

    • 一路按 D ,選 accept
    • 是否安裝最低配的 NVIDIA 驅動:
    • 是否安裝 OpenGL Libraries, 若雙顯卡選***否*** ,因爲可能會出現黑屏或者循環登錄問題
    • 是否安裝CUDA:
    • 是否鏈接 CUDA 例子:
  • 將 cuda bin 與 lib 寫進環境(.bashrc)

    • export PATH="/usr/local/cuda-9.0/bin:$PATH"
    • exprot LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"
  • reboot

驗證CUDA安裝

  • nvcc -V

CUDA 卸載

  • sudo /usr/local/cuda-9.0/bin/uninstall_cuda-9.0.pl

Nvidia驅動卸載

  • sudo /usr/bin/nvidia-uninstall

CUDNN

  • 安裝

    • 選擇合適版本下載
    • 解壓並進入相應的目錄,運行如下命令:
    sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
    sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h
    sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
    
  • 查看版本

    cat /usr/local/cuda-9.0/include/cudnn.h | grep CUDNN_MAJOR -A 2
    
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章