The command to install Nvidia GPU support on Linux machines was:
$ CONDA_OVERRIDE_CUDA=11.2 mamba install 'pytorch-gpu==cuda112'
$ CONDA_OVERRIDE_CUDA=11.2 mamba install 'pytorch-gpu=
=cuda112
'
and now it has been changed to:
$ CONDA_OVERRIDE_CUDA=11.8 mamba install 'pytorch-gpu==cuda118'
$ CONDA_OVERRIDE_CUDA=11.8 mamba install 'pytorch-gpu=
=cuda118
André Anjos authored
Lucas CARDOSO authored