zoukankan      html  css  js  c++  java
  • (原)Ubuntu安装TensorRT

    转载请注明出处:

    https://www.cnblogs.com/darkknightzh/p/11129472.html

    参考网址:

    https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html

    https://arleyzhang.github.io/articles/7f4b25ce/

    https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

    1 说明

    1.1 直接使用deb安装的。不过不记得之前cuda驱动是如何安装的了。网上说cuda驱动和TensorRTF都是deb安装的才行。。。

    1.2 安装环境是ubuntu 16.04.1+anaconda(python3.6.8)+cuda9.0+cudnn7.5

    2 安装步骤

    2.1 sudo dpkg -i nv-tensorrt-repo-ubuntu1604-cuda9.0-trt5.1.5.0-ga-20190427_1-1_amd64.deb

    2.2 sudo apt-key -add /var/nv-tensorrt-repo-cuda9.0-trt5.1.5.0-ga-20190427/7fa2af80.pub

    2.3 sudo apt-get update

    2.4 sudo apt-get install tensorrt

    至此,TensorRTF安装完成。

    2.5 通过下面命令检查一下是否安装成功:

    dpkg -l | grep TensorRT

    2.6 通过下面命令安装(不知道是啥,反正安装就对了):

    sudo apt-get install python3-libnvinfer-dev

    2.7 到上面,python里面无法import tensorrt。直接使用ubuntu自带的archive manager打开nv-tensorrt-repo-ubuntu1604-cuda9.0-trt5.1.5.0-ga-20190427_1-1_amd64.deb文件,可以看到里面有python3.6的XXX。通过下面命令安装该whl文件。

    pip install tensorrt-5.1.5.0-cp36-none-linux_x86_64.whl

    2.8 验证python能否导入tensorrt(此处成功):

    2.9 安装PyCUDA。

    PyCUDA允许python直接访问CUDA的API。

    首先保证nvcc在PATH路径中。而后输入:

    pip install 'pycuda>=2017.1.1'

    2.10 验证程序能否运行(见下面网址)

    https://arleyzhang.github.io/articles/7f4b25ce/

    可以把 tensorrt 文件夹拷贝到用户目录下,方便自己修改测试例程中的代码。

    进入 samples 文件夹直接 make,会在 bin 目录中生成可执行文件,可以一一进行测试学习。

    运行了sample_mnist,结果如下:

    3 卸载

    具体见网址:

    https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html#uninstalling

    sudo apt-get purge "libnvinfer*"
    pip uninstall tensorrt

    4 其他

    4.1 cudnn使用deb安装:

    说明:不建议使用这种方式安装。

    Navigate to your <cudnnpath> directory containing cuDNN Debian file.

    Install the runtime library, for example:

    sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb

    Install the developer library, for example:

    sudo dpkg -i libcudnn7-devel_7.0.3.11-1+cuda9.0_amd64.deb

    Install the code samples and the cuDNN Library User Guide, for example:

    sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb

    4.2 使用tar文件安装 

    Navigate to your <cudnnpath> directory containing the cuDNN Tar file.

    Unzip the cuDNN package.

    tar -xzvf cudnn-9.0-linux-x64-v7.tgz

    Copy the following files into the CUDA Toolkit directory, and change the file permissions.

    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*

    4.3 测试cudnn是否安装成功

    1)/usr/src/cudnn_samples_v7放到home文件夹下。

    2)cd mnistCUDNN/

    3)make

    4)./mnistCUDNN

    出错了。。。

    原因是之前/usr/local/cuda中cudnn用的是7.4.2版本的。替换为7.5之后,运行成功。(实际上在/usr/lib/x86_64-linux-gnu中有刚刚使用deb装上的7.5版本的cudnn,但是程序未能找到。不知道目前有两套相同版本的cudnn,以后会不会出问题吧。。。所以不建议使用deb安装,位置不可控)

  • 相关阅读:
    VirtualBox不显示64bit版本的iso
    学习和参考资料
    神经网络和机器学习资料整理
    动态空间释放时的错误操作引起的运行时错误
    WIN7 X64的运行命令窗口
    vs2010中的ADO控件及绑定控件
    AdventureWorks2012.mdf的使用
    VS2008/2010 都不能使用Access2010数据库
    WIN7 64位操作系统 无法找到Access驱动
    如何在VS2010的VC++ 基于对话框的MFC程序中添加菜单
  • 原文地址:https://www.cnblogs.com/darkknightzh/p/11129472.html
Copyright © 2011-2022 走看看