用GPU训练的模型,能部署到本地CPU环境吗?
收藏
请教:
在AIStudio上面,用GPU训练的模型,能下载到本地CPU环境,部署运行吗?
0
收藏
请登录后评论
当然可以的
GPU和CPU训练得到的模型都一样的,部署的话,一般部署说明都单独有个文档,按照文档操作就行。
谢谢解答,太好了。
我训练的是基于paddlehub的文本分类模型。哪里有部署文档?是在模型里面吗?
我在本地部署的时候,在下载ernie的时候,提示Not compiled with CUDA,是什么原因呢?
我本地就是CPU环境啊。
这个要有请几位NLP大佬回答了……
解决了。是自己的问题,我在module.py里面,写了use_cuda=True,导致出错。
改成false就好了。因为本地环境没有GPU。