在Ubuntu下配置TensorRT的步驟如下:
安裝CUDA和cuDNN:首先需要安裝CUDA和cuDNN,TensorRT依賴于這兩個軟件包。可以從NVIDIA官網下載對應版本的CUDA和cuDNN,并按照官方文檔的指導進行安裝。
下載TensorRT:從NVIDIA官網下載對應版本的TensorRT。選擇合適的版本并下載壓縮包。
解壓TensorRT壓縮包:將下載的TensorRT壓縮包解壓到合適的目錄中。
設置環境變量:打開終端,編輯.bashrc文件(或者是.zshrc文件),添加以下內容:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/TensorRT/lib
export PATH=$PATH:/path/to/TensorRT/bin
將"/path/to/TensorRT"替換為TensorRT解壓后的目錄路徑。
cd /path/to/TensorRT/samples
make
通過以上步驟,您可以在Ubuntu系統中成功配置和使用TensorRT。