• Ubuntu20.04+GTX1060+显卡驱动+CUDA11.8+cuDNN8.5.0


    注意: 不要采用之前安装在UBUNT18.04时 所用的sudo ubuntu-drivers autoinstall 命令自动安装驱动,安装后了造成内核崩溃, WIFI 适配器也无法识别!

    安装CUDA 三件套:
    1.NVIDIA GPU 驱动 (我的是GTX 1060)
    2.Cuda toolkit 11.2
    3.cuDNN (给DNN用的驱动)

    1. 安装驱动, 参考链接: https://docs.nvidia.com/datacenter/tesla/tesla-installation-notes/index.html#ubuntu-lts

    在这里插入图片描述

    $ sudo apt-get install linux-headers-$(uname -r)
    $ distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
    $ wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb
    $ sudo dpkg -i cuda-keyring_1.0-1_all.deb
    $ sudo apt-get update
    $ sudo apt-get -y install cuda-drivers
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6

    安装后检查:
    $ nvidia-smi
    在这里插入图片描述
    2. 安装CUDA Toolkit 11.2, 采用方法是 install CUDA toolkit from CUDA repository。
    先点击这里的官网,下载
    在这里插入图片描述

    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
    sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
    wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb
    sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb
    sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub
    sudo apt-get update
    sudo apt-get -y install cuda
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7

    使用命令
    $ nvcc -V
    检查是否安装成功, 如果没成功, 用以下命令添加环境变量即可:

    echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc
    source .bashrc
    
    • 1
    • 2

    以下是示列:
    在这里插入图片描述
    note: 估计官方的命令错误, 安装时命令显示的是cuda 11.2版本, 装好后成了11.8的版本。

    1. 安装cuDNN
      官网下载 : 下载时需要NVIDIA账户, 没有的话注册一个即可。
      Download cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x 下的
      Local Installer for Ubuntu20.04 x86_64 (Deb)
      下载后点击运行即可。
      在这里插入图片描述
  • 相关阅读:
    码蹄集 - MT2201 · 各位之和
    Docker认识即安装
    剑指offer62 圆圈中最后剩余的数字
    MYSQL的建表语句
    泛微 E-Office文件上传漏洞复现(CVE-2023-2523、CVE-2023-2648)
    03 Spring_DI 依赖注入
    每日一题--658. 找到 K 个最接近的元素
    iwebsec靶场 SQL注入漏洞通关笔记2- 字符型注入(宽字节注入)
    取Dataset子集(pytorch)
    html 常见兼容性问题
  • 原文地址:https://blog.csdn.net/weixin_43229238/article/details/127711920