You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
**For pytorch versions older than 1.8**, we recommend installing [torch-batch-svd](https://github.com/KinglittleQ/torch-batch-svd)
70
+
**With old pytorch versions (torch<1.8)**, we recommend installing [torch-batch-svd](https://github.com/KinglittleQ/torch-batch-svd)
71
71
to achieve a significant speed-up with `special_procrustes` on CUDA GPUs.
72
72
You can check that this module is properly loaded using the function `roma.utils.is_torch_batch_svd_available()`.
73
73
**With recent pytorch installations (torch>=1.8), `torch-batch-svd` is no longer needed or used.**
**For pytorch versions older than 1.8**, we recommend installing `torch-batch-svd <https://github.com/KinglittleQ/torch-batch-svd>`_
31
+
**With old pytorch versions (torch<1.8)**, we recommend installing `torch-batch-svd <https://github.com/KinglittleQ/torch-batch-svd>`_
32
32
to achieve a significant speed-up with :func:`~roma.mappings.procrustes` on CUDA GPUs (see section :ref:`Why a new library?`).
33
33
You can check that this module is properly loaded using the function :func:`~roma.utils.is_torch_batch_svd_available()`.
34
34
**With recent pytorch installations (torch>=1.8), torch-batch-svd is no longer needed or used.**
0 commit comments