I would be grateful for assistance installing TensorRT in to a virtual environment on a Jetson Nano B01.
I’m an amateur home user and have been working with a couple B01s since September 2021.
If I can just be pointed in the right direction, I should be able to work out the details myself.
I’d like to use a yolo3 tiny model so it seems that I need to install onnx as well so need help with that as well.
My starting point is the nvidia installation instructions found at:
If I read correctly, pip can’t be used on the Jetson Nano for installation of TensorRT. I did try it and, as expected, it did not work.
There are a few other methods listed which I have little or no experience with.
Here’s my setup:
CUDA 10.2.300
cuDNN: 8.2.1.32
Cuda ARCH: 5.3
in .bashrc:
export OPENBLAS_CORETYPE=ARMV8
PATH=“/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:/home/james/Public/Projects/myPython36VEnv/bin:/usr/local/cuda-10.2/bin:$PATH”
export CPATH=$CPATH:/usr/local/cuda-10.0/targets/aarch64-linux/include
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.0/targets/aarch64-linux/lib
export LD_LIBRARY_PATH=“/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH”
Also, I have entered these commands:
$ sudo sh -c “echo ‘/usr/local/cuda/lib64’ >> /etc/ld.so.conf.d/nvidia-tegra.conf”
$ sudo ldconfig
My virtual environment includes:
Python 3.7
numpy version 1.19.5
wheel version 0.37.1
setuptools version 49.6.0
and more that I have not listed…
I’ve also successfully installed pycuda:
pip3 install -U pycuda
→
Successfully installed appdirs-1.4.4 mako-1.2.0 platformdirs-2.5.2 pycuda-2021.1 pytools-2022.1.9
Any direction would be greatly appreciated.
James