在矩池云上复现 PaddleGAN 照片转油画风格教程

大家好，欢迎来到IT知识分享网。

我选用的是cuda10.1纯镜像

案例来自paddlepaddle官方PaddleGAN库中，可以查看

https://github.com/PaddlePaddle/PaddleGAN/

https://github.com/wzmsltw/PaintTransformer

安装需要的软件包

apt-get install cuda-cusolver-dev-10-1

conda install paddlepaddle-gpu==2.1.2 cudatoolkit=10.1 --channel https://mirrors.bfsu.edu.cn/anaconda/cloud/Paddle/

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch

一键上手

拉取并试用

git clone https://gitee.com/matpools/PaintTransformer.git
cd PaintTransformer/inference/
python inference.py

然后在inference.py文件中修改成你自己图片的名字，保存运行即可。

完成。

自己配置

也可以使用拉取官方镜像

git clone https://github.com/wzmsltw/PaintTransformer.git

下载mode，可以通过官方下载链接自己下载，然后上传到网盘。

mv /mnt/paint_best.pdparams PaintTransformer/inference/
cd PaintTransformer/inference
python inference.py

检查显存使用率

nvidia-smi -l 5

参考文章

https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/conda/linux-conda.html

https://github.com/wzmsltw/PaintTransformer

https://github.com/PaddlePaddle/PaddleGAN/blob/develop/README_cn.md