天天看點

[orin] nvidia orin 上配置tensorrt | 解決虛拟環境中路徑無法import的問題概述解決方法測試

版本:jetpack 5.0.1,tensorrt 8.4.1.5

概述

tensorrt會跟着jetpack的包一起安裝。系統本身自帶的python是3.8的版本。tensorrt的python包位于這個路徑下:

但是由于我們是有一個虛拟環境的,是以隻需要把原系統中的tensorrt元件複制過來就解決問題了。

解決方法

到系統本身的python目錄下,把

tensorrt-8.4.1.5.dist-info

tensorrt

兩個檔案夾拷貝到虛拟環境(我這裡的名字叫做

mmdeploy

)中。

tensorrt-8.4.1.5.dist-info

(72KB)和

tensorrt

(3.1MB)檔案都非常小,可以放心大膽的直接複制到虛拟環境中。

cd  /usr/lib/python3.8/dist-packages/ 
sudo cp -r tensorrt-8.4.1.5.dist-info/ ~/miniforge-pypy3/envs/mmdeploy/lib/python3.8/site-packages/tensorrt-8.4.1.5.dist-info
sudo cp -r tensorrt ~/miniforge-pypy3/envs/mmdeploy/lib/python3.8/site-packages/tensorrt
           

測試

然後測試一下是否已經解決:

[orin] nvidia orin 上配置tensorrt | 解決虛拟環境中路徑無法import的問題概述解決方法測試

可以看到已經能夠在python環境中import了!