When installing krasis using krasis-setup, pytorch will be installed using cu126 as the max option (see function _get_cuda_version_from_driver).
But for better blackwell support we need at least 12.8, which you do mention in DEV.md.
Maybe the setup.py needs an update to support 12.8 (or even 13.0) as well?
When installing krasis using krasis-setup, pytorch will be installed using cu126 as the max option (see function _get_cuda_version_from_driver).
But for better blackwell support we need at least 12.8, which you do mention in DEV.md.
Maybe the setup.py needs an update to support 12.8 (or even 13.0) as well?