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
    

你可能感兴趣的:(cudnn,CUDA)