-
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 |
|---|---|---|
| .. | ||
| __init__.py | ||
| cli.py | ||
| forecaster.py | ||
| market_data.py | ||
| metrics.py | ||
| pipeline.py | ||
| repository.py | ||
| schema.py |