Retrieve CUDA available memory via torch.cuda.mem_get_info()
#166
Job | Run time |
---|---|
11m 17s | |
11m 17s |
torch.cuda.mem_get_info()
#166
Job | Run time |
---|---|
11m 17s | |
11m 17s |