I’m having trouble getting conda to install pytorch with CUDA on WSL2. I have done the necessary setup for WSL2 on Windows 11, running Ubuntu 20.04 fully updated and the latest Nvidia WSL drivers (version 510.06, as per the Nvidia WSL website 4).
However when I try to install pytorch via conda as per the usual command
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
I keep getting the cpu-only version of pytorch.
pytorch                   1.10.0              py3.9_cpu_0    pytorch
pytorch-mutex             1.0                         cpu    pytorch
torchaudio                0.10.0                 py39_cpu  [cpuonly]  pytorch
torchvision               0.11.1                 py39_cpu  [cpuonly]  pytorch

Would anyone know how to get conda to pull the right packages for pytorch with CUDA enabled? Thanks!