API symbolscuda
Copy
tensorplay.cuda.set_device
- tensorplay.cuda.set_device(device: Any) None[source]
Set the current device.
Usage of this function is discouraged in favor of
device. In most cases it’s better to useCUDA_VISIBLE_DEVICESenvironmental variable.- Parameters:
device (tensorplay.Device or int) – selected device. This function is a no-op if this argument is negative.
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?