I’ve combed through all of the similar questions but to no avail. I have an older GPU that I’m trying to get to work with CUDA and pytorch. For some reason, I can’t get conda to install pytorch gpu. No matter what I do, it tries to install the cpu version of pytorch.

I build my conda like this – miniconda

conda create –name tortoise python=3.9 numba inflect

I need to force a specific version of CUDA 11.4.

conda install pytorch=1.11.0 torchvision=0.12 torchaudio=0.11 cudatoolkit=11.5 -c pytorch -c nvidia

Someone said that torchvision and torchaudio could cause the cpu version to be installed.

conda install pytorch=1.11.0 cudatoolkit=11.4 -c pytorch -c nvidia

Regardless of what I do, it tries to install the cpu version of pytorch.

So what it is that is determining that the cpu version should be installed. I have a GPU in this box.