PaddleHub icon indicating copy to clipboard operation
PaddleHub copied to clipboard

paddle hub 如何 调用gpu

Open terminator123 opened this issue 2 years ago • 1 comments

同时装了paddlepaddle和paddlepaddle-gpu, model = hub.Module(directory='./baidu_translate'),使用的是cpu,速度比较慢。 如果卸载paddlepaddle,只留paddlepaddle-gpu,hub会报错

File "/home/chenqun/.conda/envs/end2endtrans/lib/python3.9/site-packages/paddlehub/init.py", line 20, in _paddle_version = Version(paddle.version) AttributeError: module 'paddle' has no attribute 'version'

terminator123 avatar Feb 15 '23 03:02 terminator123

你好,看报错你应该是没有安装paddle导致的,你可以试试 import paddle 应该会报错。另外,这个模型其实是一个远程调用的模型,并不会用到本地算力

rainyfly avatar Feb 17 '23 08:02 rainyfly