diff --git a/src/taggingbackends/explorer.py b/src/taggingbackends/explorer.py index 4680ab233b0f40d822098cc615c887f2c90de4fc..c8d123adee3324458ca7e256225783ab820b6067 100644 --- a/src/taggingbackends/explorer.py +++ b/src/taggingbackends/explorer.py @@ -181,6 +181,11 @@ Cannot find any Python package in project root directory: # Julia logging elif line and line[0] in "[┌│└": print(line) + # Torch logs + elif line == ' return torch._C._cuda_getDeviceCount() > 0': + # typically follows: + # UserWarning: CUDA initialization: CUDA unknown error + print(line) # tensorflow logs elif 26 < len(line): # assume line[:26] to be e.g.: 2022-05-17 18:48:15.120981