版本: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
測試
然後測試一下是否已經解決:
![](https://img.laitimes.com/img/9ZDMuAjOiMmIsIjOiQnIsIyZuBnL3MTN0IGZmJmNzczYzkTY4YTZhRzM4kTOihDMkBTMzMzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
可以看到已經能夠在python環境中import了!