本文分享了使用StyleGAN2的实战经验，包括运行环境要求、TensorFlow版本安装、预训练模型下载、代码修改及运行过程。尽管官方建议使用16GB内存的NVIDIA GPU，但作者发现6GB的GTX 1060也可运行。此外，还介绍了StyleGAN2 Encoder的初步体验和与其他版本的对比。
StyleGAN2的代码和经过训练的模型：https://github.com/NVlabs/stylegan2
有空试用了一下，有一些很浅显的经验分享给大家：
（1）先说说运行的必备环境：
其中，比较触目惊心的是需要有16GB内存的NVIDIA GPU（即：NVIDIA Tesla V100），简单跑了一下它的“run_generator.py”程序，发现我的只有6GB内存的NVIDIA Geforce GTX 1060也可以跑起来。个人感觉，如果不去“训练模型”，只是跑一些简单的应用例子，普通的显卡也可以胜任。
（2）因为StyleGAN2需要安装定制开发的“TensorFlow ops”，所以目前只支持TensorFlow 1.14，如果你默认安装的是TensorFlow 1.15，需要重新安装1.14，例如：
conda uninstall tensorflow
conda install tensorflow=1.14
conda install tensorflow-gpu=1.14
conda install keras
（3）安装定制开发的“TensorFlow ops”，还需要C语言编译器，我的电脑是Windows10 + Visual Stu
最低0.47元/天 解锁文章
345
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
