Skip to content

Commit 070c97a

Browse files
Add instructions for GPU-enabled PyTorch to installation.md (#5600)
Fixes #5333 . ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: Dženan Zukić <dzenan.zukic@kitware.com> Co-authored-by: Mingxin Zheng <mingxinz@nvidia.com>
1 parent cc171a8 commit 070c97a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

docs/source/installation.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,11 @@ To install optional features such as handling the NIfTI files using
2626
Ignite](https://pytorch.org/ignite/), please follow the instructions:
2727
- [Installing the recommended dependencies](#installing-the-recommended-dependencies)
2828

29+
The installation commands bellow usually end up installing CPU variant of PyTorch. To install GPU-enabled PyTorch:
30+
1. Install the latest NVIDIA driver.
31+
1. Check [PyTorch Official Guide](https://pytorch.org/get-started/locally/) for the recommended CUDA versions. For Pip package, the user needs to download the CUDA manually, install it on the system, and ensure CUDA_PATH is set properly.
32+
1. Continue to follow the guide and install PyTorch.
33+
1. Install MONAI using one the ways described below.
2934
---
3035

3136

0 commit comments

Comments
 (0)