I had pytorch installed in a conda environment using the CPU. On a new machine I have a GPU available, so I want to upgrade to the CUDA version. I’ve removed pytorch from the environment and then run the recommended line, in my case:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
However, I keep getting offered the CPU build:
How can I upgrade to the CUDA build?