• xeron56's avatar
    Add CUDA device compatibility checks to TimesFM backend · 6132f5c0
    xeron56 authored
    - Implemented a function to check if the current CUDA device is supported by the installed PyTorch version.
    - Added warnings for unsupported CUDA devices, informing users that the CPU will be used instead.
    - Updated the TimesFMBackend class to utilize the new compatibility checks and handle GPU names appropriately.
    - Introduced unit tests to validate the behavior of the CUDA compatibility function with various simulated CUDA environments.
    6132f5c0
timesfm-prediction.component.html 9.45 KB