转TensorRT

https://github.com/NVIDIA-AI-IOT/torch2trt

pip3 install tensorrt-5.0.2.6-py2.py3-none-any.whl  
# pycuda
pip3 install pycuda 
发布了2674 篇原创文章 · 获赞 982 · 访问量 528万+

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/104543467