We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 281fbfa commit 46b1797Copy full SHA for 46b1797
tests/core/test_index_utils.py
@@ -2,6 +2,7 @@
2
3
# First Party
4
from smdebug.core.index_reader import S3IndexReader
5
+from smdebug.core.logger import get_logger
6
from smdebug.core.s3_utils import list_s3_objects
7
from smdebug.core.utils import (
8
deserialize_tf_device,
@@ -15,10 +16,14 @@
15
16
17
18
def test_tf_device_name_serialize_and_deserialize():
19
+ logger = get_logger()
20
import tensorflow.compat.v1 as tf
21
22
device_name = tf.test.gpu_device_name()
23
if not bool(device_name):
24
+ logger.warning(
25
+ "There is no GPU Support on this machine. Please ignore the cuInit errors generated above"
26
+ )
27
device_name = "/device:GPU:0"
28
29
serialized_device_name = serialize_tf_device(device_name)
0 commit comments