基于tensorflow2 的WGAN实现以及模型结构、损失函数的个人理解总结

---

---

1. 环境安装

1.1 环境搭建

创建一个tensorflow2的新环境，这样一是方便安装避免与之前安装的包冲突，二是可以方便管理代码，避免以后各种包之间依赖关系错误，无法安装。

conda create -n tensorflow2 -python=3.7.0

1.2 GPU是否可用检测

激活环境，输入pip命令安装tensorflow2

pip install tensorflow==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

检测一下，GPU版本是否可用

import tensorflow as tf
tf.test.is_gpu_available()

如果输出类似如下结果，就表明GPU确实可用，关键是最后那个tr