可以直接用libtorch部署python的.pt文件吗？

水木社区手机版

主题:可以直接用libtorch部署python的.pt文件吗？
楼主|wakaka1122|2024-11-19 23:04:48|只看此ID
比如把python模型转成.pt文件，然后用libtorch直接部署到生产环境，这样可以吗？
--
FROM 112.64.60.*
1楼|grainbuds|2024-11-22 11:16:28|只看此ID
可以的，英伟达显卡的话也可以考虑tensorrt
--
FROM 39.144.103.*