-
Notifications
You must be signed in to change notification settings - Fork 45.4k
Closed
Description
when running ptb_word_lm.py with tf-nightly 1.5.0-dev20171103
I got this error
2017-11-04 16:48:59.759628: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1121] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1070, pci bus id: 0000:01:00.0, compute capability: 6.1)
2017-11-04 16:48:59.761131: E tensorflow/stream_executor/cuda/cuda_driver.cc:936] failed to allocate 221.62M (232390656 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
WARNING:tensorflow:From /home/larry/workspace/tensorflow/models/tutorials/rnn/ptb/ptb_word_lm.py:165: get_or_create_global_step (from tensorflow.contrib.framework.python.ops.variables) is deprecated and will be removed in a future version.
Instructions for updating:
Please switch to tf.train.get_or_create_global_step
Traceback (most recent call last):
File "/home/larry/workspace/tensorflow/models/tutorials/rnn/ptb/ptb_word_lm.py", line 527, in <module>
tf.app.run()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "/home/larry/workspace/tensorflow/models/tutorials/rnn/ptb/ptb_word_lm.py", line 502, in main
model.import_ops()
File "/home/larry/workspace/tensorflow/models/tutorials/rnn/ptb/ptb_word_lm.py", line 272, in import_ops
params_saveable = tf.contrib.cudnn_rnn.RNNParamsSaveable(
AttributeError: 'module' object has no attribute 'RNNParamsSaveable'
It seems that there is no RNNParamsSaveable in cudnn_rnn
Any idea how to fix it?
Metadata
Metadata
Assignees
Labels
No labels