Skip to content

Commit

Permalink
Merge pull request #8003 from leofang/cut_docs
Browse files Browse the repository at this point in the history
Update installation guide to note about cuTENSOR 2.0 support
  • Loading branch information
emcastillo committed Nov 29, 2023
2 parents 08409a7 + 32b2d23 commit f6b5622
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions docs/source/install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ Additional CUDA Libraries

Part of the CUDA features in CuPy will be activated only when the corresponding libraries are installed.

* `cuTENSOR <https://developer.nvidia.com/cutensor>`_: v1.6 / v1.7
* `cuTENSOR <https://developer.nvidia.com/cutensor>`_: v2.0

* The library to accelerate tensor operations. See :doc:`../reference/environment` for the details.

Expand Down Expand Up @@ -81,12 +81,6 @@ Package names are different depending on your CUDA Toolkit version.

* - CUDA
- Command
* - **v10.2** (x86_64 / aarch64)
- ``pip install cupy-cuda102``
* - **v11.0** (x86_64)
- ``pip install cupy-cuda110``
* - **v11.1** (x86_64)
- ``pip install cupy-cuda111``
* - **v11.2 ~ 11.8** (x86_64 / aarch64)
- ``pip install cupy-cuda11x``
* - **v12.x** (x86_64 / aarch64)
Expand Down

0 comments on commit f6b5622

Please sign in to comment.