diff --git a/doc/install.rst b/doc/install.rst index 7826c8047002f036082ecc1a1e03858729ec3b6c..e07e4e4587a7d36a2b9685f8d30eb2d3e89a6006 100644 --- a/doc/install.rst +++ b/doc/install.rst @@ -45,6 +45,18 @@ We support two installation modes, through pip_, or mamba_ (conda). mamba install -c https://www.idiap.ch/software/biosignal/conda/label/beta -c conda-forge ptbench + .. tip:: + + To force-install Nvidia GPU support on Linux machines, execute: + + .. code:: sh + + $ mamba install pytorch-gpu + + # or, to force the Nvidia CUDA version (environments w/o Nvidia setup): + + $ CONDA_OVERRIDE_CUDA=11.2 mamba install 'pytorch-gpu=*=cuda112*' + .. _ptbench.setup: