Jetson Xavier NX中自带TensorRT,cuda,cudnn

cuda位置在/usr/local/cuda
cudnn和TensorRT的.so文件位置在/usr/lib/aarch64-linux-gnu/
TensorRT位置在/usr/src/tensorrt

cuda

nvcc -V

cudnn

cd /usr/src/cudnn_samples_v8/mnistCUDNN/
sudo make 
./mnistCUDNN

cudnn运行结果

TensorRT
进入下载mnist数据的目录

cd /usr/src/tensorrt/data/mnist

运行下载测试数据

将download_pgms.py复制出来到home文件夹下
python3 ./download_pgms.py
sudo cp 0.pgm /usr/src/tensorrt/data/mnist

进入sample_mnist 的目录并编译

cd /usr/src/tensorrt/samples/sampleMNIST
sudo make

运行sample_mnist

cd ../../bin/
./sample_mnist

在这里插入图片描述

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐