SRGAN-tensorflow icon indicating copy to clipboard operation
SRGAN-tensorflow copied to clipboard

CUDA_ERROR_OUT_OF_MEMORY Failed to create session

Open Cybrak opened this issue 7 years ago • 0 comments
trafficstars

Hello! I am trying to run the test_SRGAN.sh but I am getting an error. tensorflow.python.framework.errors_impl.InternalError: Failed to create session.

I am running tensor flow 1.4.1 in virtualenv

Any ideas where its coming from and how to fix it ?

Thanks ! Full Error

E tensorflow/core/common_runtime/direct_session.cc:170] Internal: failed initializing StreamExecutor for CUDA device ordinal 0: Internal: failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_OUT_OF_MEMORY; total memory reported: 8499691520
Traceback (most recent call last):
  File "main.py", line 20, in <module>
    sess = tf.Session(config = config)
  File "/home/cybrak/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1482, in __init__
    super(Session, self).__init__(target, graph, config=config)
  File "/home/cybrak/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 622, in __init__
    self._session = tf_session.TF_NewDeprecatedSession(opts, status)
  File "/home/cybrak/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/errors_impl.py", line 473, in __exit__
    c_api.TF_GetCode(self.status.status))
tensorflow.python.framework.errors_impl.InternalError: Failed to create session.

Cybrak avatar Mar 29 '18 06:03 Cybrak