From 85e82c3d12ac12e8a2e1a707e1380ac22c995450 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 24 Oct 2023 07:51:01 +0200
Subject: [PATCH] [doc] Update installation instructions to include (mamba)
pytorch-gpu installation tips
---
doc/install.rst | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/doc/install.rst b/doc/install.rst
index 7826c804..e07e4e45 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:
--
GitLab