TensorRT5在Ubuntu16.04中的開發環境搭建

TensorRT5在Ubuntu16.04中的開發環境搭建

一、Ubuntu16.04系統安裝及準備

1、Ubuntu16.04系統安裝

參考:https://blog.csdn.net/weixin_40494464/article/details/81010256比較簡單

這個教程比較靠譜,安裝多了基本就這樣咯,很簡單就不重複贅述了。

2、安裝好以後的準備(基本的工具)

(1)sougou輸入法安裝

(2)tmux安裝

(3)cmake安裝:https://cmake.org/download/ 下載linux版本,

源碼安裝

       1)卸載舊版本

      sudo apt-get autoremove cmake

       2) 解壓安裝

      $ tar zxvf cmake-3.13.3.tar.gz
      $ cd cmake-3.13.3
      $ ./configure
      $ make
      $ sudo make install

      3)查看版本

       $ cmake –version

      上面是源碼安裝現在講一下,載預編譯文件安裝,下載cmake-3.13.3-Linux-x86_64.tar.gz版本,

      1tar zxvf cmake-3.13.3-Linux-x86_64.tar.gz

      2mv cmake-3.13.3-Linux-x86_64 /opt/cmake-3.13.3

      3ln -sf /opt/cmake-3.13.3/bin/*  /usr/bin/  (創建軟連接)

(4)qt安裝

(5)vim安裝 sudo apt-get install vim

查看當前系統內核版本:uname -r

二、顯卡驅動安裝

1、查看ubuntu16.04默認顯卡驅動(安裝前需要禁用該顯卡驅動)

lsmod | grep nouveau

禁用:1)  sudo gedit /etc/modprobe.d/blacklist.conf ,隨後添加下面兩行並保存:

        blacklist nouveau

options nouveau modeset=0

      2)sudo update-initramfs -u 然後重啓

        然後執行驗證是否禁用成功: lsmod | grep nouveau,如果沒有輸出信息就代表禁用成功。

2、下載顯卡驅動進入網址:選擇符合自己電腦配置的顯卡驅動下載

https://www.nvidia.com/Download/index.aspx?lang=en-us

安裝過程:

  1. ubuntu系統下按住 ctrl+alt+F1進入命令行
  2. sudo service lightdm stop 關閉圖形界面
  3. sudo apt-get remove nvidia-*卸載之前版本
  4. 進入下載好的顯卡驅動所在目錄 sudo chmod a+x NVIDIA-Linux-x86_64-410.93.run
  5. Sudo ./ NVIDIA-Linux-x86_64-410.93.run -no-x-check -no-nouveau-check -no-opengl-files
  1. The distribution-provided pre-install script failed! Are you sure you want to continue? 選擇 yes 繼續。
  2. Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?  選擇 No 繼續。
  3. 問題沒記住,選項是:install without signing
  4. 問題大概是:Nvidia's 32-bit compatibility libraries? 選擇 No 繼續。
  5. Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  選擇 Yes  繼續

以上內容摘抄自:

https://blog.csdn.net/xunan003/article/details/81665835

  1. 安裝完畢之後nvidia-smi查看,顯示顯卡狀態則代表安裝成功

出現的問題:使用一段時間突然提示錯誤:NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the ...: 解決就是重新安裝驅動,而不用動cuda, 這樣也非常快!

 

三、cuda9.0+、cudnn7.3和TensorRT安裝

一)、cuda9.0

1、去官網https://developer.nvidia.com/cuda-downloads下載cuda9.0

 

下載的版本爲:cuda_9.0.176_384.81_linux.run

2、進入安裝文件所在目錄:

1)sudo chmod 777 cuda_9.0.176_384.81_linux.run

2)sudo ./ cuda_9.0.176_384.81_linux.run --no-opengl-libs,提示是否安裝驅動384.81則選擇否。

3)sudo gedit ~/.bashrc 行尾部加入:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

 

source ~/,bashrc  (更新文件)

4)驗證cuda是否安裝成功

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

sudo make

sudo ./deviceQuery

補充操作:

卸載cudasudo /usr/local/cuda-9.0/bin/uninstall_cuda-9.0.pl

卸載NVIDIA Driver :   sudo  /usr/bin/nvidia-uninstall

  3、系統中安裝了多個版本的cuda,則切換只需要修改一下軟連接:

1) sudo gedit ~/.bashrc 

修改爲:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

2) sudo /etc/profile    #必須更改/etc/profile 文件, 而且更改後必須重啓計算機纔有效 (source /etc/profile 不能生效

export PATH=/usr/local/cuda/bin:$PATH 

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

3) 切換

cuda9.0切換到cuda8.0

sudo rm -rf /usr/local/cuda                       #刪除之前創建的軟鏈接sudo ln -s /usr/local/cuda-8.0  /usr/local/cuda   #創建新 cuda 的軟鏈接

 

cuda8.0切換到cuda9.0

sudo rm -rf /usr/local/cuda                       #刪除之前創建的軟鏈接sudo ln -s /usr/local/cuda-9.0  /usr/local/cuda   #創建新 cuda 的軟鏈接

二)、cudnn 7.3.0

chmod -R 777 filename #文件及子目錄權限改變

去官網下載:https://developer.nvidia.com/rdp/cudnn-download

版本:cudnn7.3.0

 

 

解壓下載好的文件: sudo tar -zxvf cudnn_name

複製文件到cuda9.0下面:

cd cuda

sudo cp include/cudnn.h /usr/local/cuda-9.0/include/

sudo cp lib64/* /usr/local/cuda-9.0/lib64/

三)、TensorRT5

   去官網:https://developer.nvidia.com/tensorrt

   下載最新版本TensorRT5.0.2.6

解壓:sudo tar -zxvf xxxx.tar.gz

配置環境變量:

Sudo vim ~/.bashrc

添加環境變量:export LD_LIBRARY_PATH= TensorRT解壓路徑/lib:${LD_LIBRARY_PATH:+:$ {LD_LIBRARY_PATH}}

其實在開發中只需要將TensorRT的lib及include配置到CmakeLists.txt中即可。

四、Opencv3.4.2+安裝及配置(或者4.0)

   待續...

五、Anaconda安裝及配置

 待續....

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章