wc-cuda 0.1.3

No project description provided

wc_cuda

A wrapper around windows-capture that captures the screen directly into PyTorch CUDA tensors, avoiding CPU memory copies and inter-device memory transfers.

To avoid reinstalling torch, it's not listed as a dependency, but a CUDA-enabled PyTorch installation is required. It only works on x64 Windows with an NVIDIA GPU.

High-FPS mode (optional)

WindowsCapture accepts:

• reuse_output_buffer (default: False)
• output_buffer_count (default: 3)

Usage

from wc_cuda import WindowsCapture

capture = WindowsCapture(
    monitor_index=1,
    device_id=0,
    reuse_output_buffer=True,
    output_buffer_count=4,
)

When buffer reuse is enabled, tensors may be overwritten after several frames (ring-buffer behavior), so clone tensors in your callback if you need to keep them long-term.

Credits

This was written by gemini-cli and Claude Code. Originally written by NiiightmareXD windows-capture-python and forked from Nagadomi wc_cuda, then further optimized by Claude Code.