Skip to content

Commit 46b1797

Browse files
authored
Explain cuInit Error Logs (aws#226)
1 parent 281fbfa commit 46b1797

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/core/test_index_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
# First Party
44
from smdebug.core.index_reader import S3IndexReader
5+
from smdebug.core.logger import get_logger
56
from smdebug.core.s3_utils import list_s3_objects
67
from smdebug.core.utils import (
78
deserialize_tf_device,
@@ -15,10 +16,14 @@
1516

1617

1718
def test_tf_device_name_serialize_and_deserialize():
19+
logger = get_logger()
1820
import tensorflow.compat.v1 as tf
1921

2022
device_name = tf.test.gpu_device_name()
2123
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+
)
2227
device_name = "/device:GPU:0"
2328

2429
serialized_device_name = serialize_tf_device(device_name)

0 commit comments

Comments
 (0)