rknn-toolkit装了好多次，总是会报错，终于尝试出了一个版本可以用的。（注意windows和linux上都能用）

rknn-toolkit: v1.7.1：
cuda 10.1

python==3.6
pytorch==1.5.1
opencv-python==3.2.0.8
tensorflow==1.14.0

模型导出的时候，最好pytorch版本也是1.5.1，导出的代码是：

trace_model = torch.jit.trace(model.model, torch.rand(1, 3, 32, 38).cuda())
torch.jit.save(trace_model, "./rknn_ocr.pt")

如果不行的话自己先试一试，可视化里面导入的代码是：

model = torch.jit.load('./rknn_ocr.pt')