本次的实验环境:
- LINUX(Ubuntu 16.04)
- python3.6
- Nvidia GPU 1080 Ti
- Cuda 9.0
- Cudnn 7.0.4
安装:
1、安装Pytorch 0.4
conda install pytorch=0.4 cuda90 -c pythorch
2、安装visdom and dominate
pip install visdom dominate
3、下载github中cycleGAN模型文件夹
git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
cd pytorch-CycleGAN-and-pix2pix
4、安装所有的依赖
pip install -r requirements.txt
5、下载数据集
bash ./datasets/download_cyclegan_dataset.sh maps
或者点击taesung_park网址下载所需数据集(教育网址)