Hello, I used 2080ti, tensorflow-gpu 1.12.0, CUDA 9.0, cudnn7.6.5 to reproduce the code test part, but I encountered a bug in the middle.
I found that the reason was that running on 2080ti would crash, so I used CPU to reproduce, Step 1 of test part can run correctly, but there is a problem when running step 2。
WARNING:tensorflow:From /root/anaconda3/envs/python3.6/lib/python3.6/site-packages/tflearn/initializations.py:119: UniformUnitScaling.__init__ (from tensorflow.python.ops.init_ops) is deprecated and will be removed in a future version.
Instructions for updating:
Use tf.initializers.variance_scaling instead with distribution=uniform to get equivalent behavior.
Traceback (most recent call last):
File "/root/anaconda3/envs/python3.6/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 1628, in _create_c_op
c_op = c_api.TF_FinishOperation(op_desc)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be rank 3 but is rank 4 for 'graph_localproj_1_layer_1/Pad' (op: 'Pad') with input shapes: [3,224,224,16], [3,2].
Would this problem only appear in the CPU version and not in the GPU version? Do you have a good solution?
Hello, I used 2080ti, tensorflow-gpu 1.12.0, CUDA 9.0, cudnn7.6.5 to reproduce the code test part, but I encountered a bug in the middle.
I found that the reason was that running on 2080ti would crash, so I used CPU to reproduce, Step 1 of test part can run correctly, but there is a problem when running step 2。
Would this problem only appear in the CPU version and not in the GPU version? Do you have a good solution?