我选用的是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://github.com/wzmsltw/PaintTransformer
https://github.com/PaddlePaddle/PaddleGAN/blob/develop/README_cn.md