Skip to content

Commit 859c6bc

Browse files
author
Löffler, Hannes
committed
clearer description of available GPUs
1 parent e0bd3b3 commit 859c6bc

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ Installation
5151
conda create --name reinvent4 python=3.10
5252
conda activate reinvent4
5353
```
54-
1. Change directory to the repository and install all dependencies. You will need to set the right processor type, see [PyTorch versions](https://pytorch.org/get-started/locally/). Linux supports CUDA (e.g. "cu124"), ROCm (e.g. "rocm6.2.4") and CPU. Windows supports CUDA and CPU. MacOSX only supports CPU (use "mac" as processor type!). Optionally, you can select dependencies "openeye" (for ROCS; you need to obtain your own license), "isim" for similarity tracking in TensorBoard or "none" to skip all. The default is installation of "all" dependencies. See the help text from the install script for details.
54+
1. Change directory to the repository and install all dependencies. You will need to set the right processor type, see [PyTorch versions](https://pytorch.org/get-started/locally/). Linux supports Nvidia CUDA (e.g. "cu124"), AMD ROCm (e.g. "rocm6.2.4") and CPU. Windows supports CUDA and CPU. MacOSX only supports CPU (use "mac" as processor type!). Optionally, you can select dependencies "openeye" (for ROCS; you need to obtain your own license), "isim" for similarity tracking in TensorBoard or "none" to skip all. The default is installation of "all" dependencies. See the help text from the install script for details.
5555
```shell
5656
python install.py --help
5757
python install.py cu124 # or rocm6.2.4, cpu, mac, etc.

0 commit comments

Comments
 (0)