Skip to content

Commit cbebbad

Browse files
author
Simon Rit
committed
DOC: Installation instructions for the CUDA-compatible Python package
1 parent 21dd15e commit cbebbad

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

INSTALLATION.md

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,14 +26,18 @@ Installation is currently not supported for independent RTK compilations.
2626

2727
Pre-compiled binaries
2828
---------------------
29-
We only provide pre-compiled binaries for the Python package which depends on ITK. Use the following commands to install RTK module with `pip`.
29+
We only provide pre-compiled binaries for the Python package which depends on ITK. Use the following commands to install the RTK module with `pip`.
3030
```
3131
python -m pip install --upgrade pip
3232
python -m pip install itk-rtk
3333
```
34-
3534
The same operating systems and Python versions are supported as ITK's packages, see the list on [Pypi](https://pypi.org/project/itk-rtk).
3635

36+
We also provide pre-compiled [CUDA](https://developer.nvidia.com/cuda-toolkit) packages for Windows and Linux. They require an installed version of CUDA 11.6. Use the following command instead of the above one to install the CUDA RTK module with `pip`.
37+
```
38+
python -m pip install itk-rtk-cuda116
39+
```
40+
3741
Getting started
3842
---------------
3943
See [GettingStarted.md](GettingStarted.md). Your `CMakeLists.txt` can now use RTK when importing ITK as shown in the [FirstReconstruction's CMakeLists.txt](https://github.com/SimonRit/RTK/blob/master/examples/FirstReconstruction/CMakeLists.txt#L7).

0 commit comments

Comments
 (0)