Skip to content

Commit

Permalink
comment
Browse files Browse the repository at this point in the history
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
tensorflow/tensorflow#24496
  • Loading branch information
albertz authored and Spotlight0xff committed Sep 5, 2020
1 parent 384e4a8 commit 16624c8
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions tests/test_TFNetworkLayer.py
@@ -1,11 +1,23 @@

# start: nosetests $this_file --nologcapture
from __future__ import division

import logging
logging.getLogger('tensorflow').disabled = True

import os
# Get us some further useful debug messages (in some cases, e.g. CUDA).
# For example: https://github.com/tensorflow/tensorflow/issues/24496
# os.environ["CUDNN_LOGINFO_DBG"] = "1"
# os.environ["CUDNN_LOGDEST_DBG"] = "stdout"
# The following might fix (workaround): Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
# (https://github.com/tensorflow/tensorflow/issues/24496).
# os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"

import tensorflow as tf
import sys
sys.path += ["."] # Python 3 hack

from nose.tools import assert_equal, assert_is_instance
import contextlib
import unittest
Expand Down

0 comments on commit 16624c8

Please sign in to comment.