draugr.torch_utilities.system.device¶
Created on 15/11/2019
Functions
|
first time call stores to device for global reference, later call must explicitly manually override! |
|
|
|
|
Auto selects the device with highest compute capability and with the requested memory available |
|
|
|
Destroy cuda state by emptying cache and collecting IPC. |
Classes
|
An enumeration. |