File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -140,11 +140,11 @@ def env():
140
140
cuda_table = Table (show_header = False , show_lines = False )
141
141
cuda_table .add_row ("CUDA available" , str (torch .cuda .is_available ()))
142
142
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 ) )
144
144
cuda_table .add_row (
145
145
"cuDNN version" , format_cudnn_version (torch .backends .cudnn .version ())
146
146
)
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 () ))
148
148
cuda_table .add_row ("GPU type" , torch .cuda .get_device_name ())
149
149
total_memory_gb = float (torch .cuda .mem_get_info ()[1 ]) / float (
150
150
1024 * 1024 * 1024
You can’t perform that action at this time.
0 commit comments