How to get PyTorch to recognize gpus when using Anaconda

Technical Help
2

There is no need to downgrade

  1. Create a new env:
    conda create -n pytorch python=3.8
  2. enter the env
    conda activate pytorch
  3. install pytorch as in docs
    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
1 Like