TensorFlow 中遇到的坑

最近在做GANS的实验，用到了TensorFlow，把坑总结一下。

1 ResourceExhaustedError

具体是这样的：现在主机上有四块卡，有一块有任务。
这也是TensorFlow很坑爹的一块：训练程序默认吃掉所有的显存。用掉所有的卡。。。

解决办法

#CUDA_VISIBLE_DEVICES=1 python my_script.py
设置这个训练程序只可以看到一块卡。卡号为1
或者设置如下，
#CUDA_VISIBLE_DEVICES=0,1 python demo.py

2 ‘module’ object has no attribute ‘pack’

#老版本的TensorFlow代码会这样#如output_shape=tf.pack([tf.shape(out_1)[0], IMAGE_SIZE//8, IMAGE_SIZE//8, 256]), strides=[1, 2, 2, 1], padding='SAME')AttributeError: 'module' object has no attribute 'pack'

解决：因为TF后面的版本修改了这个函数的名称，把tf.pack改为 tf.stack。