Skip to content

Commit 86c3f5b

Browse files
modified instalation with pypi to warn about cupy on rocm
1 parent c742550 commit 86c3f5b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: docs/source/installation/installation_pypi.rst

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ A GPU-supported version of the tool can be installed directly from PyPI using th
2121
Note that the ``cuda12x`` can be exchanged by:
2222

2323
- ``cuda``: Triggers the non-compiled version of ``cupy`` so that it builds specifically for your system.
24-
- ``rocm-4-0``: Deploys a version of ``cupy`` tailored for AMD GPUs using ROCM.
24+
- ``rocm-4-0``: in order to deploy a version of ``cupy`` tailored for AMD GPUs using ROCM. ROCM is still experimental. Please be aware that the CUDA/ROCM version should match the ones in your system.
25+
2526

2627
Optional Modules
2728
----------------

0 commit comments

Comments
 (0)