• Ubuntu18.04上安装N卡驱动、CUDA、CUDNN三连


    环境:Ubuntu18.04

    显卡驱动真的挺方便的,CUDA和CUDNN还是踩了一些坑2333

    1、安装显卡驱动
    安装ubuntu更新或sudo apt-get update & sudo apt-get upgrade

    sudo add-apt-repository ppa:graphics-drivers/ppa
    sudo apt-get update
    ubuntu-drivers devices
    sudo ubuntu-drivers autoinstall

    安装显卡驱动

    sudo reboot
    nvidia-smi

    若输出显卡信息则安装成功

    非常方便了

    2、安装CUDA
    在官网https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal下载安装包,默认下载到Downloads目录下

    cd ~/Downloads/

    运行下载的.run文件,如:cuda_10.1.105_418.39_linux.run

    sudo sh cuda_*_linux.run

    在过程中,选择需要安装的内容时,请将驱动driver去掉,不要安装显卡驱动!

    配置环境变量(该步骤似乎并不需要,没有进行该操作也成功运行了后续测试,有待考证)

    export CUDA_HOME=/usr/local/cuda 
    export PATH=$PATH:$CUDA_HOME/bin 
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    source ~/.bashrc
    sudo reboot

    测试是否安装成功,这里如果没有重启过,貌似运行的时候会有报错

    cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
    sudo make
    ./deviceQuery

    查看CUDA版本

    cat /usr/local/cuda/version.txt

    3、安装CUDNN
    在官网https://developer.nvidia.com/cudnn注册账号并下载,选择下载cuDNN Library for Linux,不过说实话现在官网给的版本好多好多看花了眼23333,默认下载到Downloads目录下,解压并复制

    cd ~/Downloads/
    tar -xzvf cudnn-*
    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 /usr/local/cuda/lib64/libcudnn*

    查看CUDNN版本

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
  • 相关阅读:
    python-深浅copy-18
    Python-集合-17
    linux-阿里云仓库搭建-搭建本地仓库-yum
    python-知识回顾-16
    python-编码-15
    python-小知识点-14
    codevs 1048石子归并
    codevs 1048 石子归并
    codevs1068乌龟棋
    codevs 1697 ⑨要写信
  • 原文地址:https://www.cnblogs.com/cenariusxz/p/10841628.html
Copyright © 2020-2023  润新知