Created
February 28, 2019 19:12
-
-
Save vsuthichai/6b6b028c43a3c29e9568723518283641 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| diff --git a/tensorflow/python/keras/backend.py b/tensorflow/python/keras/backend.py | |
| index fc2e8aa..e3081c0 100644 | |
| --- a/tensorflow/python/keras/backend.py | |
| +++ b/tensorflow/python/keras/backend.py | |
| @@ -594,7 +594,8 @@ def _has_nchw_support(): | |
| bool: if the current scope device placement would support nchw | |
| """ | |
| explicitly_on_cpu = _is_current_explicit_device('CPU') | |
| - gpus_available = bool(_get_available_gpus()) | |
| + #gpus_available = bool(_get_available_gpus()) | |
| + gpus_available = True | |
| return not explicitly_on_cpu and gpus_available | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment