You've already forked FFmpeg
mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2026-04-24 04:44:54 +02:00
a077da895b
Add support for CUDA and ROCm (AMD GPU) devices in the LibTorch DNN backend. This works for both NVIDIA CUDA and AMD ROCm, as PyTorch exposes ROCm through the CUDA-compatible API. Usage: ./ffmpeg -i input.mp4 -vf scale=224:224,format=rgb24,dnn_processing=dnn_backend=torch:model=sr_model_torch.pt:device=cuda output.mp4 Reviewed-by: Guo Yejun <yejun.guo@intel.com> Signed-off-by: younengxiao <steven.xiao@amd.com>