• Ubuntu16.04安装cuda10.2 + cuDNN 7.6.5.32 (包含Nvidia驱动)


     

    此经验帖参考了CUDA官网教程cuDNN官网安装教程师兄经验帖,经测试适用于惠普暗影精灵笔记本(Nvidia GTX 1660 Ti 显卡) & 组装台式机(Nvidia RTX 2080 Ti 显卡).

    1、CUDA 10.2安装(包含Nvidia显卡驱动)

    如果Ubuntu系统下没有装过Nvidia显卡驱动,不必单独安装,直接装CUDA即可自动安装显卡驱动.

    (1)注意先看自己显卡型号是不是支持CUDA
    查看显卡型号的命令:
    $ lspci | grep -i nvidia
    支持CUDA的Nvidia显卡型号列表

    (2)如果显卡支持,就可以去官网下载CUDA,选Ubuntu16.04对应的runfile(local)文件:
    CUDA 10.2官网下载地址 默认下载到Home目录下

    (3)去home目录下找到自己下载的CUDA的run文件名字,如cuda_10.2.89_440.33.01_linux.run,记下来一会要用

    (4)命令行模式下安装CUDA
    先 Ctrl+Alt+F1 进入命令行模式:
    $ sudo service lightdm stop
    $ sudo sh ./cuda_10.2.89_440.33.01_linux.run
    按照它要求的选择accept,install,耐心等它装完
    $ sudo service lightdm start

    (5)重新进入图形界面后,配置环境文件
    $ sudo gedit ~/.bashrc
    在末尾加上以下内容:

    export PATH=/usr/local/cuda-10.2/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
    
    • 1
    • 2

    保存退出
    $ source ~/.bashrc

    检查安装情况:
    nvidia-smi 检查GPU,nvcc -V 检查CUDA

    更新:
    尝试用 runfile 方式在 Ubuntu 18.04下装CUDA 10.2,安装失败,改用deb方式安装成功。

    方法:去官网下载deb(local)并参照官网说明安装即可:
    https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal

    2、cuDNN 7.6.5.32安装

    cuDNN是深度学习一般要用到的,取需

    (1)官网下载
    cuDNN 7.6.5.32官网下载地址
    注册后下载这三个东西,其实就是对应于ubuntu16.04的deb文件都下载下来:
    cuDNN Runtime Library for Ubuntu16.04 (Deb)
    cuDNN Developer Library for Ubuntu16.04 (Deb)
    cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)

    (2)之后完全按照官网教程安装和测试即可

    3、跑mnist测试时可能会遇到问题

    给两台电脑装了cuDNN,其中一台跑mnist测试时报错:
    bash: ./mnistCUDNN: No such file or director

    我的解决办法是把官网安装教程里从 2.3.2. Installing From A Debian File 到 2.4. Verifying The cuDNN Install On Linux 的所有过程重新来一遍,就解决了。

  • 相关阅读:
    drf之版本控制
    drf自定义筛选
    drf----分页
    初识drf
    flask源码分析01:框架简介
    Werkzeug之LocalStack源码解析
    Werkzeug之Local源码解析
    threading.local之数据存储隔离
    python之attr
    Python在pycharm中编程时应该注意的问题汇总
  • 原文地址:https://www.cnblogs.com/leoking01/p/13657560.html
Copyright © 2020-2023  润新知