PaddleGAN icon indicating copy to clipboard operation
PaddleGAN copied to clipboard

Published 20 hours ago • verified publisher icon PaddlePaddle

please change CUDAPlace(0) to be CPUPlace

Open KiisMyGun opened this issue 3 years ago • 1 comments

项目：AnimeGANv2 CPU：i7-6700K 执行：python tools/main.py --config-file configs/animeganv2_pretrain.yaml 错误：Cannot use GPU because you have installed CPU version PaddlePaddle. If you want to use GPU, please try to install GPU version PaddlePaddle by: pip install paddlepaddle-gpu If you only have CPU, please change CUDAPlace(0) to be CPUPlace() 怎么切换呢？在main.py中加入import paddle paddle.set_device('cpu') 不得行

KiisMyGun avatar Mar 06 '21 06:03 KiisMyGun

@KiisMyGun

错误原因是您安装的不是GPU版本的PaddlePaddle.

AnimeGANv2模型预测，设置--cpu即可。

https://github.com/PaddlePaddle/PaddleGAN/blob/develop/applications/tools/animeganv2.py#L33

qingqing01 avatar Mar 08 '21 03:03 qingqing01

问题过于久远，如果有图像和视频生成的需求，可以使用新的跨模态工具: https://github.com/PaddlePaddle/PaddleMIX/tree/develop

JunnYu avatar Feb 29 '24 03:02 JunnYu