Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
Romain BRÉGIER committed Jul 9, 2024
1 parent 4af5c29 commit fb8dadf
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.**
Expand Down
2 changes: 1 addition & 1 deletion docsource/source/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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 <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 :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.**
Expand Down

0 comments on commit fb8dadf

Please sign in to comment.