Skip to content
Snippets Groups Projects
Commit b12418be authored by Jan Hartig's avatar Jan Hartig
Browse files

Log number of used GPUs

parent a1f69625
Branches
No related tags found
No related merge requests found
...@@ -24,6 +24,8 @@ if len(capableGPUs) < 1: ...@@ -24,6 +24,8 @@ if len(capableGPUs) < 1:
print("No {} capable cuda device found!".format(compute_type)) print("No {} capable cuda device found!".format(compute_type))
exit(1) exit(1)
print("Using {} {} capable GPUs".format(len(capableGPUs), compute_type))
tStart = perf_counter() tStart = perf_counter()
# Run on GPU with FP32 # Run on GPU with FP32
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment