• ubuntu20.04显卡1080ti安装cuda和cudnn


    0Ubuntu升级内核后显卡丢失的处理

    (1)先删除显卡驱动 

    sudo apt-get purge nvidia* 

    (2)再自动安装

    sudo ubuntu-drivers autoinstall

    (3)再重启

    sudo reboot

    -----------------------------------------------------------------------------


     

    1/参考了

    https://blog.csdn.net/weixin_37926734/article/details/123033286

    2、显卡驱动安装

    (1)安装驱动

    sudo ubuntu-drivers autoinstall

    3、安装cuda

    (2)安装cuda可以用迅雷下载了再安装

    wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run

    sudo sh cuda_11.7.0_515.43.04_linux.run

    参照最上面的教程进行安装,安装后进行配置

    CUDA安装完成后,需要配置变量环境才能正常使用。首先在终端输入sudo gedit ~/.bashrc打开如下图所示的.bashrc文件。
    然后,如下图所示在.bashrc文件的最后添加以下CUDA环境变量配置信息

    sudo gedit ~/.bashrc

    在最后面输入

    1. export PATH=$PATH:/usr/local/cuda/bin
    2. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
    3. export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

    最后输入

    source ~/.bashrc

    而后输入下面命令查看是否安装成功

    1. nvcc -V
    2. #查看CUDA的安装信息

    4、安装cudnn

    https://developer.nvidia.com/rdp/cudnn-archive

    下载了.deb的包(上面2个包都可以用迅雷下载)

    sudo dpkg -i cudnn-local-repo-ubuntu2004-8.4.1.50_1.0-1_amd64.deb

    正在选中未选择的软件包 cudnn-local-repo-ubuntu2004-8.4.1.50。
    (正在读取数据库 ... 系统当前共安装有 204785 个文件和目录。)
    准备解压 cudnn-local-repo-ubuntu2004-8.4.1.50_1.0-1_amd64.deb  ...
    正在解压 cudnn-local-repo-ubuntu2004-8.4.1.50 (1.0-1) ...
    正在设置 cudnn-local-repo-ubuntu2004-8.4.1.50 (1.0-1) ...

    The public CUDA GPG key does not appear to be installed.
    To install the key, run this command:
    sudo cp /var/cudnn-local-repo-ubuntu2004-8.4.1.50/cudnn-local-E3EC4A60-keyring.gpg /usr/share/keyrings/

    sudo cp /var/cudnn-local-repo-ubuntu2004-8.4.1.50/cudnn-local-E3EC4A60-keyring.gpg /usr/share/keyrings/

    而后到/var/cudnn-xxx目录下面

    1. sudo dpkg -i libcudnn8_8.4.1.50-1+cuda11.6_amd64.deb
    2. sudo dpkg -i libcudnn8-dev_8.4.1.50-1+cuda11.6_amd64.deb
    3. sudo dpkg -i libcudnn8-samples_8.4.1.50-1+cuda11.6_amd64.deb

    先安装以下依赖,再测试以下是否安装成功了,否则可能提示: cudnn  fatal error: FreeImage.h: 没有那个文件或目录

    sudo apt-get install libfreeimage3 libfreeimage-dev

    而后进入/usr/src/cudnn_samples_v8/mnistCUDNN/ 中

    1. sudo make clean
    2. sudo make

    最后测试以下

    ./mnistCUDNN

    出现 Test passed! 代表 cudnn 正常使用

    5、安装tensorrt

    (1)可以迅雷下载自己想要的

    https://developer.nvidia.com/nvidia-tensorrt-8x-download

    TensorRT-8.4.3.1.Linux.x86_64-gnu.cuda-11.6.cudnn8.4.tar.gz 

     (2)在home下面建一个tensorrt目录以后就不动了,先解压

    tar xzvf TensorRT-${version}.Linux.${arch}-gnu.${cuda}.${cudnn}.tar.gz

    (3)根据自己的真实情况然后添加环境变量:设置路径

    1. sudo gedit ~/.bashrc
    2. # 换成你自己的路径
    3. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<TensorRT-${version}/lib>
    4. 如:
    5. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/chen/tensorrt/TensorRT-8.4.3.1/lib
    6. # 保存关闭
    7. source ~/.bashrc

    (4)先安装anaconda然后创建虚拟环境,再安装下面的操作。

    主要参考了

    https://blog.csdn.net/laizi_laizi/article/details/121567609

    因为主要之后用的是python API,然后选择你虚拟环境对应的python版本的用pip安装:

    另外两个包也可以安装一下:

    1. cd TensorRT-${version}/graphsurgeon
    2. python3 -m pip install graphsurgeon-0.4.5-py2.py3-none-any.whl
    1. cd TensorRT-${version}/onnx_graphsurgeon
    2. python3 -m pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl


    通常xxx\TensorRT-x.x.x.x\下也有uff,是用于转化tensorflow的模型的,不用tf的话就不用安装了

    (5)测试
    上面都完事了可以编译测试一下:
    我这边因为是最新版的trt,所以里面已经有了测试图片(以.pgm结尾的,所以也没有download_pgms.py文件了):

    只需要简单的切换到samples/sampleMNIST下make一下就行

    (open-mmlab) lz@lz:~/Downloads/TensorRT-8.2.1.8/samples/sampleMNIST$ make
    1. ~/tensorrt/TensorRT-8.4.3.1/samples/sampleMNIST$
    2. 下面使用
    3. make

    然后返回bin目录,运行一下就通过测试了

    1. cd ../../bin/
    2. ./sample_mnist

  • 相关阅读:
    设计模式-单例模式 (Singleton)
    并发编程之 ThreadLocal
    Shell 脚本面试指南
    redis源码阅读-zset
    Modbus协议介绍及Modbus TCP
    Ribbon 客户端负载均衡
    Vue实现模糊查询:filter()
    VScode远程连接Linux入门
    第一百五十二回 自定义组件综合实例:游戏摇杆三
    framework源码读后感
  • 原文地址:https://blog.csdn.net/chencaw/article/details/127121904