TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.set_per_process_memory_fraction

tensorplay.cuda.set_per_process_memory_fraction(fraction, device: Any | None = None) → None[source]

Set memory fraction for a process.

Not enforced by this TensorPlay build; validated for signature parity.

Parameters:

  • fraction (float) – Range: 0~1. Allowed memory equals total_memory * fraction.

  • device (tensorplay.Device or int, optional) – selected device.

tensorplay.cuda.set_device

Set the current device.

tensorplay.cuda.set_stream

Usage of this function is discouraged in favor of the stream context manager.

Ask DeepWiki