-
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
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| agents | ||
| api | ||
| dataflows | ||
| dgt_forecasting | ||
| graph | ||
| llm_clients | ||
| timesfm_forecasting | ||
| __init__.py | ||
| dashboard.py | ||
| default_config.py | ||
| languages.py | ||
| reporting.py |