diff --git a/README.md b/README.md index 214ec28..3b08f4c 100644 --- a/README.md +++ b/README.md @@ -67,7 +67,7 @@ Alternatively one can install the latest version of *RoMa* directly from the sou pip install git+https://github.com/naver/roma ``` -**For pytorch versions older than 1.8**, we recommend installing [torch-batch-svd](https://github.com/KinglittleQ/torch-batch-svd) +**With old pytorch versions (torch<1.8)**, we recommend installing [torch-batch-svd](https://github.com/KinglittleQ/torch-batch-svd) to achieve a significant speed-up with `special_procrustes` on CUDA GPUs. You can check that this module is properly loaded using the function `roma.utils.is_torch_batch_svd_available()`. **With recent pytorch installations (torch>=1.8), `torch-batch-svd` is no longer needed or used.** diff --git a/docsource/source/index.rst b/docsource/source/index.rst index baa7003..4bb3b56 100644 --- a/docsource/source/index.rst +++ b/docsource/source/index.rst @@ -28,7 +28,7 @@ Alternatively one can install the latest version of *RoMa* directly from the sou pip install git+https://github.com/naver/roma -**For pytorch versions older than 1.8**, we recommend installing `torch-batch-svd `_ +**With old pytorch versions (torch<1.8)**, we recommend installing `torch-batch-svd `_ to achieve a significant speed-up with :func:`~roma.mappings.procrustes` on CUDA GPUs (see section :ref:`Why a new library?`). You can check that this module is properly loaded using the function :func:`~roma.utils.is_torch_batch_svd_available()`. **With recent pytorch installations (torch>=1.8), torch-batch-svd is no longer needed or used.**