Skip to content

Commit bc60996

Browse files
committed
update installation instructions
1 parent 4fc1205 commit bc60996

2 files changed

Lines changed: 14 additions & 7 deletions

File tree

README.md

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,15 @@ latest version using pip:
1414

1515
> pip install renard-pipeline
1616
17-
If you have a GPU, there are accelerated versions for Nvidia CUDA and
18-
AMD ROCm:
17+
By default, this pulls the cuda version of pytorch. If you want a CPU-only version of torch you should do:
1918

20-
> pip install renard-pipeline[cuda128]
21-
> pip install renard-pipeline[rocm63]
19+
> pip install torch --index-url https://download.pytorch.org/whl/cpu
20+
> pip install renard-pipeline
21+
22+
For an AMD rocm version of torch:
23+
24+
> pip install torch --index-url https://download.pytorch.org/whl/rocm6.4
25+
> pip install renard-pipeline
2226
2327

2428
# Documentation

docs/installation.rst

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,13 @@ Installation
55
Using Pip
66
=========
77

8-
For the simplest case, use ``pip install renard-pipeline``. By default, this installs the CPU version of PyTorch. If you want GPU support to accelerate inference:
8+
For the simplest case, use ``pip install renard-pipeline``. By default, this installs the cuda version of PyTorch. For other versions:
99

10-
- CUDA 12.8: ``pip install renard-pipeline[cuda128]``
11-
- ROCm 6.3: ``pip install renard-pipeline[rocm63]``
10+
CPU only:
11+
``pip install torch --index-url https://download.pytorch.org/whl/cpu && pip install renard-pipeline``
12+
13+
ROCM 6.4:
14+
``pip install torch --index-url https://download.pytorch.org/whl/rocm6.4 && pip install renard-pipeline``
1215

1316

1417
Note that for some modules, you might need to install additional

0 commit comments

Comments
 (0)