Tensorrt環境搭建

2021-10-07 16:15:37 字數 1577 閱讀 6147

python 3.6

tensorrt 7.0.0.1

cd /path/to/tensorrt7.0.0.1


pip install tensorrt-7.0.0.11-cp36-none-linux_x86_64.whl
pytorch 1.5

wget 


sudo sh cuda_10.2.89_440.33.01_linux.run
編輯~/.bashrc

export path=$path:/usr/local/cuda-10.2/bin


export ld_library_path=$ld_library_path:/usr/local/cuda-10.2.0/lib64
cuda安裝注意事項一定不要安裝預設的驅動,而且options裡面也不需要編譯opengl的庫

nvidia驅動安裝

pycuda

pip install  pycuda -i   --trusted-host mirrors.aliyun.com
pycharmopencv

pip install  opencv-python==3.4.2.17 -i   --trusted-host mirrors.aliyun.com
nvidia-docker安裝src/cpp/cuda.hpp:14:18: fatal error: cuda.h: no such file or directory

可能是沒有安裝cuda或者是安裝cuda後include沒有加入path

pycharm伺服器執行沒有圖形介面,加入-x

ssh -x [email protected]
pip install老是安裝到其他路徑了,可以指定安裝的位置

sudo ~/anaconda3/bin/python setup.py install
4.檢視顯示卡使用情況

watch nvidia-smi
cuda gpu算力對照表:tensorrt c++原始碼庫位址

sudo cmake .. -dtrt_lib_dir=$trt_release/lib -dtrt_bin_dir=`pwd`/out -dcmake_cuda_compiler:path=/usr/local/cuda/bin/nvcc

VS2017配置Tensorrt的環境

1 安裝英偉達的驅動 跟以前一樣 2 安裝cuda10的版本，過程中不要選精簡，選自定義然後全選。3 安裝配置cudnn7.6.3 4 安裝vs2017 5 配置tensorrt推導的環境 確保自己已經安裝好cuda，沒有cuda的就不要繼續了。新建專案 選擇nvidia cuda xx，選擇自己名...

LAMP環境搭建 php環境搭建

yum groupinstall development tools y yum groupinstall desktop platform development y yum install cmake pcre devel ncurses devel openssl devel libcurl ...

環境搭建 二 Python環境搭建

工欲善其事，必先利其器 要想玩轉python，我們首先需要搭建一套可使用且方便的環境。對於初學者，笨飯糰建議使用 anaconda anaconda 簡介anaconda是python的乙個科學計算發行版，內建了數百個python經常會使用的庫，包括做機器學習或資料探勘的庫。anaconda提供了乙...