diff --git a/doc/install.rst b/doc/install.rst index aaf5baaae32b3bdf6319edc2c31e393ef3e7c75e..2f14ba7a1824b4ac52f04b37d682563e6803a7cd 100644 --- a/doc/install.rst +++ b/doc/install.rst @@ -55,7 +55,7 @@ We support two installation modes, through pip_, or mamba_ (conda). # or, to force the Nvidia CUDA version (environments w/o Nvidia setup): - $ CONDA_OVERRIDE_CUDA=11.2 mamba install 'pytorch-gpu=*=cuda112*' + $ CONDA_OVERRIDE_CUDA=12.0 mamba install 'pytorch-gpu=*=cuda118*' .. _mednet.setup: