DCGAN

1、代码获取https://github.com/carpedm20/DCGAN-tensorflow
2、数据集获取
python download.py mnist celebA存在问题,不能下载
直接从官网http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html下载数据集,然后解压放到data文件夹下(文件夹内有20多万张图片)
在这里插入图片描述
3、代码运行环境Ubuntu Server 16.04 x64 + cuda8,利用conda进行管理,python3.5,tensorflow-gpu1.4.0

需解决的问题:
1、gpu的选择
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘1’ //1代表第二块gpu
2、代码运行过程中出现错误tensorflow.python.framework.errors_impl.InternalError,from device: CUDA_ERROR_OUT_OF_MEMORY
1)先运行nvidia-smi 检查GPU运行情况,若内存够用进入2
2)代码应作已下修改

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0' #use GPU with ID=0
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 # maximun alloc gpu50% of MEM
config.gpu_options.allow_growth = True #allocate dynamically
sess = tf.Session(config = config)

3、出现如下错误
在这里插入图片描述
在main.py代码开头加上下面两行:

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

4、出现如下错误
在这里插入图片描述
解决方法:
(1)tensorflow(gpu版本)的版本是1.10.0,此版本需要cuda9.0+;与服务器上cuda8.0+版本不匹配
(2)采用了降低tensorflow版本(<tensorflow1.4.0)的方法,安装tensorflow1.4.0(pip install tensorflow-gpu==1.4.0)

代码改动
在这里插入图片描述

参考:
[1]https://blog.csdn.net/Candy_GL/article/details/81138297
[2]https://blog.csdn.net/sinat_30372583/article/details/79461468
[3]https://blog.csdn.net/zhaohaibo_/article/details/80573676
[4]https://blog.csdn.net/Alawaka2018/article/details/80301703
[5]https://blog.csdn.net/u012436149/article/details/53837651

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值