Skip to content
Snippets Groups Projects
Commit 85e82c3d authored by André Anjos's avatar André Anjos :speech_balloon: Committed by Daniel CARRON
Browse files

[doc] Update installation instructions to include (mamba) pytorch-gpu installation tips

parent 5ce011d2
No related branches found
No related tags found
1 merge request!12Adds grad-cam support on classifiers
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment