SRGAN-tensorflow
SRGAN-tensorflow copied to clipboard
CUDA_ERROR_OUT_OF_MEMORY Failed to create session
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.