vm.torch.cuda¶

Functions¶

current_device(…) : Return the index of current selected device.

get_device_capability(…) : Return the capability of specified device.

get_device_name(…) : Return the name of specified device.

is_available(…) : Return a bool reporting if runtime is available.

set_device(…) : Set the current device.

synchronize(…) : Synchronize all streams on a device.