- 博客(2)
- 收藏
- 关注
原创 Jeston Nx使用Tensor RT 加速神经网络步骤
1.准备好要用的文件包,tensorrt x(由国外大神制作的tensorrt 的接口),yolo v5 源码,注意,yolo v5 源码的版本一定要与tensorrt x 的版本对应。yolo v5训练的权重。5.在自己的深度学习环境里,也就是第3步的终端对话框里,输入命令。3.打开终端激活自己的深度学习环境,这一步默认使用本教程的人已经装好了anaconda、python、pytorch等环境,没装的可以自行百度。打开终端,在终端中输入命令nvcc -V,若出现以下界面就说明cuda已经安装完成。
2023-07-13 17:57:02
985
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅