文章目录
jetson系列编译部署libtorch
安装
- 查看设备
cuda版本
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0
-
官网下载对应编译库Start Locally | PyTorch，jetson系列无法使用，会提示版本不符合，需要自己编译
-
根据Jetpack版本，去论坛找相应源码[PyTorch for Jetson - version 1.9.0 now available]
-
安装依赖环境
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
- 编译pytorch
sudo pip3 install torch-1.9.0-cp36-cp36m-linux_aarch64.whl
- 查看安装版本
pip3 list | grep torch
模型转换
- pytorch模型转化为libtorch的torchscript模型(.pth—>.pt)
def convert_model(model, path_out, img_tensor, running_mode='gpu'):
if running_mode == 'gpu' and torch.cuda.is_available():
print("gpu")
device = torch.device("cuda:0")
model = model.cuda(device) # mdoel.to(device)
model.load_state_dict(torch.load(path_model), strict=False)
img_tensor = img_tensor.to(device)
else:
print("cpu")
# device = torch.dev
780
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
