- 博客(2)
- 收藏
- 关注
原创 第四章 Jetson Nano Unet TensorRT模型推理
使用Unet模型训练好的.pth模型文件利用onnx库转化为.onnx模型文件,然后利用onnx-tensorrt将onnx模型文件转换为tensorrt的engine文件即.trt模型文件,最终使用tensorrt进行推理
2022-06-09 16:18:19
3302
1
原创 Pytorch环境搭建
pytorch环境搭建1. conda create -n env_name python=3.82. conda activate env_name3. conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
2022-03-30 18:17:20
2016
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人