-
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