Skip to content

Commit d25d4a8

Browse files
authored
Fix printing errors in oumi env for non-string values. (#1565)
1 parent 6f6e869 commit d25d4a8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/oumi/cli/env.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -140,11 +140,11 @@ def env():
140140
cuda_table = Table(show_header=False, show_lines=False)
141141
cuda_table.add_row("CUDA available", str(torch.cuda.is_available()))
142142
if torch.cuda.is_available():
143-
cuda_table.add_row("CUDA version", torch.version.cuda)
143+
cuda_table.add_row("CUDA version", str(torch.version.cuda))
144144
cuda_table.add_row(
145145
"cuDNN version", format_cudnn_version(torch.backends.cudnn.version())
146146
)
147-
cuda_table.add_row("Number of GPUs", torch.cuda.device_count())
147+
cuda_table.add_row("Number of GPUs", str(torch.cuda.device_count()))
148148
cuda_table.add_row("GPU type", torch.cuda.get_device_name())
149149
total_memory_gb = float(torch.cuda.mem_get_info()[1]) / float(
150150
1024 * 1024 * 1024

0 commit comments

Comments
 (0)